sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
223 stars 77 forks source link

For training on custom dataset, I saw that in your smoke.yaml example stated that object class need to +1, does that mean if my dataset got 28classes originally, then I also need to +1 classes, which make my number of classes become 29? #11

Closed MheadHero closed 2 years ago

MheadHero commented 2 years ago

One more extra question is that does this repo require any specific dataset format? Like in the example is voc format?

sovit-123 commented 2 years ago

Yes, it is required VOC format. It may support more formats in the future. But it is a complicated thing to handle and may require more time.