search

thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

How can I train with my own dataset? help wanted #1126

Closed Mikkochu closed 4 years ago

Mikkochu commented 4 years ago

I'm trying to train darkflow with my own dataset.

I have installed darkflow, created test images and made annotations (Pascal VOC & YOLO). If I have understood correctly the darkflow training requires Pascal VOC?

My problem is that I don't know how to start the training. Am I supposed to get weights as a result of training?

ankitAMD commented 4 years ago

Change Class name in labels.txt file. Creating XML Annotations file of all images separately which class they belong. You have lot of options of configuration and weight file. like yolo.cfg , tiny-yolo.cfg. Changes in some code. Follow this github and read it. https://github.com/ankitAMD/Darkflow-object-detection

Mikkochu commented 4 years ago

Change Class name in labels.txt file. Creating XML Annotations file of all images separately which class they belong. You have lot of options of configuration and weight file. like yolo.cfg , tiny-yolo.cfg. Changes in some code. Follow this github and read it. https://github.com/ankitAMD/Darkflow-object-detection

Thank you so much. I will read the tutorial! I managed to complete the training, but now I'm having problems with testing. https://github.com/thtrieu/darkflow/issues/1127

ankitAMD commented 4 years ago

Please close this issue if you have completed.