Training on your own dataset

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.14k stars 2.08k forks source link

Training on your own dataset #1011

Open hakS07 opened 5 years ago

hakS07 commented 5 years ago

I'm trying to follow this tuto for traing yolov2 on my dataset(1 classe) https://modelzoo.co/model/yolo-tensorflow using this `flow --model cfg/yolov2-new.cfg --load bin/yolov2.weights --train --annotation train/Annotations --dataset train/Images the annotations(.txt) are in yolo format Capture du 2019-03-27 17-48-00 Capture du 2019-03-27 17-48-13 Capture du 2019-03-27 17-50-59 But i got this error

batch_per_epoch = int(size / batch) ZeroDivisionError: division by zero Dataset size: 0

skofot commented 5 years ago

I believe Darkflow requires Pascal VOC format for the annotations. LabelImg has support for producing Pascal VOC annotations and that is what I've used for mine.

S0soo commented 5 years ago

You need to convert. TXT to. XML

hakS07 commented 5 years ago

@skofot @S0soo thank you, it resolves the problem