Решение хакатона WildHack - кейс от Кроноцкого заповедника по распознаванию животных с фотоловушек. От команды НеКамчатские зайцы (December 2021)

Решение заняло 1 место. Изначально решение от команды было представлено тут: https://github.com/sweetlhare/rescue-Kamchatka-animals

Проблема, которую необходимо было решить, - с помощью искусственного интеллекта отличать фотографии, на которых отсутствуют животные или кадры плохого качества. После этого необходимо было детектировать основных животных на картинках или видео с фотоловушек.

Решение основано на использовании модели yolov5, обученной на размеченных данных, пользовательский интерфейс реализован с использованием streamlit.

Для запуска: Файл main.py нужно запустить командой streamlit run main.py

В папке models лежат веса обученной модели. В папке notebooks лежат ноутбуки с разработкой модели.