Как я могу тренироваться с собственным набором данных с Darkflow?

Я новичок с опытом программирования. Я пытаюсь обучить darkflow на собственном наборе данных. Я следую этим инструкциям.

https://github.com/thtrieu/darkflow

Пока что я сделал следующие шаги.

  • установлен darkflow и соответствующие модули

  • Создал тестовые изображения и сделал аннотации (Pascal VOC).

               https://ibb.co/y4HmtGz  

           https://ibb.co/GkxLshK

Если я правильно понял, для обучения darkflow требуется Pascal VOC?

Моя проблема в том, что я не знаю, с чего начать обучение. Как начать процесс обучения и как проверить, работает ли нейросеть? Должен ли я набирать вес в результате тренировок?


yolo darknet darkflow
person PythonBeginner__    schedule 14.01.2020    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Вы можете выбрать использование предварительно натренированных весов здесь. Загрузите cfg и weights.

Предполагая, что у вас установлен darkflow, вы можете обучить свою сеть следующим образом:

flow --model cfg/<your-config-filename>.cfg --load bin/<filename>.weights --train --annotation train/Annotations --dataset train/Images --epoch 100 --gpu 1.0

Если вы хотите обучить свою сеть с нуля без использования каких-либо предварительно обученных весов, вы можете сделать это:

flow --model cfg/<your-config-filename>.cfg --train --annotation train/Annotations --dataset train/Images --epoch 100 --gpu 1.0

После начала обучения контрольные точки модели сохраняются в каталоге ckpt. Вы можете загрузить последнюю контрольную точку и протестировать на образцах изображений.

person anisbhsl    schedule 30.03.2020