Skip to content

Latest commit

 

History

History
11 lines (6 loc) · 1.43 KB

README.md

File metadata and controls

11 lines (6 loc) · 1.43 KB

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

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

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

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

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

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