Skip to content

MaxTube-dot/Animals-Solution-

Repository files navigation

Animals-Solution

Формирование регистраций животных

Описание

Animals-Solution - это система, с помощью которой осуществляется распознавание, регистрация животных для последующего использования в деятельности неинвазивного мониторинга объектов животного мира.

Скринкаст

train_batch2521

Разработка

При разработке системы были использованы такие технологии, как:

  • Python
  • YOLOv8n
  • TKinter
  • PyTorch

Описание структуры baseline решения

configs/config.py - файл, содержащий конфигурационные классы с полями
configs/config.yml - файл формата yml, который содержит значения параметров по умолчанию для детектора, классифкатора и т.д.

data - директория с данными для отладки работы системы

utils/utils.py - файл содержит функции для подгрузки классификатора, детектора, файла с названиями классов и другие вспомогательные функции

weights - директория, которая содержит веса модели детектора YOLOv8 размера nano и классификатора efficientnet_b0.

class_map_animal.txt - файл с названиями классов, на которых был обучен классификатор

main.py - файл, UI приложения для автоматической регистрации животных с камер с фотоловушек

algorithm.py - файл, содержищий алгоритм для поиска регистраций среди фотографий с фотоловушек

model.py - файл, реалзиующий инференс двухстадийной системы для нахождения и распознавания животных с фотоловушек

Инструкция по запуску

Рекомендованная версия среды Python 3.8.19

Установка библиотек к проекту

pip install -r requirements.txt

Запуск инференса

python3 main.py или python main.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages