How can I organize my own dataset for 1 class like voc?

[shonehang]

I have trained “yolo_v3_mobilenetv2_voc” and “ssd_lite_mobilenetv2_train_voc” successfully. So I want to train my own data, just 1 class: car, I have prepared “Annotations” and “JPEGImages”. I have changed NUM_CLASSES to 2 in yml file and VOC_CLASSES (...) to VOC_CLASSES = ( 'car', 'background') in voc.py. But I have no idea about txt files in ImageSets/Main, now I just have a trainval.txt which just contains the ids of the images. so I have the error: Dimension out of range (expected to be in range of [-1,0), but got 1) What additional txt file do I need?What else do I need to modify? Thank you very much.

[shonehang]

I have solved this problem. Just switch VOC_CLASSES = ( 'car', 'background') to VOC_CLASSES = ( 'background','car')

[rw1995]

can you give me some help? I load voc2007 in ssds.pytorch/data file . but when I do python train.py --cfg=./experiments/cfgs/fssd_vgg16_train_voc.yml . something is wrong

Traceback (most recent call last): File "train.py", line 20, in from lib.ssds_train import train_model File "/home/rw/ssds.pytorch/ssds.pytorch-master/lib/ssds_train.py", line 23, in from lib.dataset.dataset_factory import load_data File "/home/rw/ssds.pytorch/ssds.pytorch-master/lib/dataset/dataset_factory.py", line 2, in from lib.dataset import coco File "/home/rw/ssds.pytorch/ssds.pytorch-master/lib/dataset/coco.py", line 13, in from lib.utils.pycocotools.coco import COCO File "/home/rw/ssds.pytorch/ssds.pytorch-master/lib/utils/pycocotools/coco.py", line 55, in from . import mask as maskUtils File "/home/rw/ssds.pytorch/ssds.pytorch-master/lib/utils/pycocotools/mask.py", line 3, in import lib.utils.pycocotools._mask as _mask ImportError: No module named 'lib.utils.pycocotools._mask'

which python is yours? and how to train ?

[rw1995]

I have solved this problem. Just switch VOC_CLASSES = ( 'car', 'background') to VOC_CLASSES = ( 'background','car')

give me some reminds