File size: 1,981 Bytes
149c596
 
f8551a7
149c596
 
8bfd0e0
f8551a7
149c596
4d43fa4
149c596
2b4f19d
4adbe70
 
071b34d
 
333b7d2
a84908d
071b34d
 
a84908d
 
071b34d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
---
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