ограничивающие рамки darkflow плохо даже в стандартном примере

Question

Нейронная сеть на Darkflow не обучается правильно.

Сначала я подумал, что мне не хватает изображений или, возможно, плохая стартовая сеть.Поэтому я попробовал стандартный пример, такой как https://github.com/deep-diver/Soccer-Ball-Detection-YOLOv2/blob/master/YOLOv2-Train.ipynb, но даже тогда у меня возникла та же проблема (см. Изображение ошибки).

Я могу протестировать с предварительно обученной сетью просто отлично.Только когда я пытаюсь тренироваться в сети, я получаю ерунду.И все же в конце 100 эпох у меня есть:

шаг 1100 - потеря 1.004814863204956 - потеря движущегося аве 1.1848642304486146 Окончание 100 эпох (ов)

Я тренируюсь с:

потоком--model cfg / yolo_custom.cfg - загрузить bin / yolo.weights --train - аннотация labelImg / annotations --dataset labelImg / images --epoch 100 --gpu 1.0

и тестировать с помощью:

flow --model cfg / yolo_custom.cfg - загрузить bin / yolo.weights --demo test_video.mp4 --saveVideo

(который можно найти на этом сайте)

Он показывает изображения, которые он получает на этом сайте.Я вместо этого получаю:

https://imgur.com/a/wcdeDuY

Answer 1

Разобрался. было 2 вещи. 1. Мне нужно было сказать ему загрузить конкретную контрольную точку этой конфигурации с «--load 21000» (или с любым последним номером) 2. Мне нужно было сначала наложить его на 3-5 изображений, а затем позволить тренироваться на остальных изображениях.