/hacks-ai-walruses

Our solution to the Hacks-AI (Khabarovsk, 2022, May 27-29) competition in the task of detecting walruses from drone photos

Primary LanguageJupyter Notebook

hacks-ai-walruses

Хакатон "Цифровой прорыв"

Кейс: "ИИ на страже популяции ненецких моржей". Хабаровск, 2022.05.27-29
Команда PytBools:

  • Алёшин Максим hairymax
  • Владимир Хомяков rozendetr
  • Александр Белозёров Belka8

Задача

Разработать программный модуль учета количества моржей на лежбищах на основе данных, полученных с беспилотных летательных аппаратов на территории заповедника «Ненецкий»

Запуск обучения

Обучение нейросети для детекции и сегментации проводилось в Google Colab pro c использованием фреймворка MMDetection
train_mask_rcnn.ipynb

Установка зависимостей

Используемый стек

  • MMDedection
  • MMCV
  • PyTorch
  • Dash
  • OpenCV

В файле requirements.txt находится список необходимых библиотек
Описание устновки mmdetection и mmcv можно найти на офциальном сайте

Веса обученной нейросети

https://drive.google.com/file/d/1-GD7yanzZ3HfQ0flyqG50GRfNUemwVjY/view?usp=sharing

Запуск вебинтерфейса

python3 app/app.py

Пример работы вебинтерфейса

Pytbools_screencast_2022-05-29_720.mp4

Решение

Решение команды Pytbools на тестовых изображениях представлен в файле pytbools.zip