yolo2-pytorch icon indicating copy to clipboard operation
yolo2-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon longcw

could you please write a train demo?

Open gittigxuy opened this issue 6 years ago • 2 comments

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?

gittigxuy avatar Apr 20 '18 14:04 gittigxuy

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

jinyu121 avatar Apr 26 '18 01:04 jinyu121

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

chenruishuang avatar Jul 26 '18 13:07 chenruishuang