search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

could you please write a train demo? #80

Open gittigxuy opened 6 years ago

gittigxuy commented 6 years ago

According to the train.py file,I could know that :If I want to train my own data,I should write a class such as VOCDataset,but how can I do specifically?

jinyu121 commented 6 years ago

The easy way is that you can make a VOC-like dataset... Just organize your files as the format of VOC . Then, copy VOCDataset, change class names-- that's all.

---- Update We should alco change class names in https://github.com/longcw/yolo2-pytorch/blob/master/cfgs/config_voc.py#L8

chenruishuang commented 5 years ago

i copy the format of VOC and change the class names ,but it occurred index error . how can i solve it?