После кастомной модели YOLOv3 нет ограничительной рамки - PullRequest
0 голосов
/ 24 апреля 2019

enter image description here Я пытался обучить модель YOLO v3 для обнаружения мотоциклов. Модель хорошо тренируется и достигает средней потери 0,00001 за 1000 шагов. Но когда я проверяю это, я не вижу никаких рамок или предсказаний.

Набор данных мотоциклов - 300 изображений

файл конфигурации - tiny-yolo.cfg

классы - 1

фильтры - 30

#Traning
#batch=64
#subdivisions=8
width=416
height=416
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1

learning_rate=0.001
max_batches = 2000
policy=steps
steps=-1,100,800,1600
scales=.1,10,.1,.1

Я попробовал следующее обнаружение, оно работает хорошо, но не показывает прогноз. Скажите, пожалуйста, какие изменения я должен внести?

...