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

Question

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

https://github.com/thtrieu/darkflow

До сих пор я проделал следующие шаги.

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

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

             https://ibb.co/y4HmtGz  
  
             https://ibb.co/GkxLshK
  

Если я правильно понял, что обучение темному потоку требует Pascal VO C?

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

Answer 1

Вы можете использовать предварительно обученные веса от здесь . Скачать 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. Вы можете загрузить последнюю контрольную точку и проверить образцы изображений.