search

dshahrokhian / YOLO_tensorflow

:oncoming_taxi: Object Detection (YOLOv1) implentation in tensorflow, with training, testing and video features.
42 stars 22 forks source link

How to train with own dataset? #6

Closed Heidisnaps closed 7 years ago

Heidisnaps commented 7 years ago

Hi.everyone.

I want to use this code to do yolo training. What do I need to train my own dataset? (ex. xxx.weight, xxxx.ckpt ..etc...) And what code should I run?

Thanks:)

dshahrokhian commented 7 years ago

Hello,

The easiest solution is to structure your dataset similarly to VOC, and then change this line to reference to your own dataset directory.

Being honest, if I was you I would use this repo instead. It is much better, since ours is already outdated.

Dan

StefaniaTp commented 6 years ago

@Heidisnaps did you solve this problem? I want to try this network with my own weights and dataset, but I don't know how to do this