README.md

---
title: GrechnikNet
emoji: 🌾
colorFrom: indigo
colorTo: purple
sdk: docker
app_port: 7860
pinned: false
short_description: Detection of impurities in buckwheat
---
Наведи камеру на гречку и ищи примеси!  
Регулируй степень уверенности (conf) того, что объект является примесью и допустимую степень пересечения детектированных объектов (iou)!!  
Тебе потребуется предоставить доступ к камере!!!


Данная нейронка получена дообучением нейросети YOLO8nano из билиотеки ultralytics (одноэтапного алгоритма детекции - на выходе получает тензоры, содержащие координаты bounding боксов, степень уверенности того, что в них находится объект и условные вероятности того, что в bounding box объект относится к тому или иному классу).
Были заморожены 10 первых модулей (самых глубоких, модуль включает много слоёв) и дообучены оставшиеся 23-10+1=14 модулей.  
Класс один - примесь. Из-за по сути соединения двух классов - черных штук и некрасивых крупинок немного просел recall (метрика качества выявления положительного признака), но остался высоким precion. И в целом нейронка отлично выявляет черные штуки в гречке.  
Обучение завершилось на 15 эпохе.  
Метрики:  
P          R         mAP50     mAP50-95  
0.89      0.682      0.798      0.413