search

yjh0410 / PyTorch_YOLO-Family

Apache License 2.0
158 stars 47 forks source link

Using different dataset #11

Closed kulkarnikeerti closed 2 years ago

kulkarnikeerti commented 2 years ago

Hi @yjh0410

If I would want to use a different dataset instead of COCO and VOC, I would have to change the data pre-processing and also the data loader right? Do I have to keep anything else in mind? Or what exactly I should change? Or can I just use another dataset (ex: Mapillary) directly?

yjh0410 commented 2 years ago

@kulkarnikeerti If you want to use other dataset, you need to write a new dataset file for the new dataset. Please refer to voc.py for the format of the file content. As for other data pre-processing like data augmentation and data loader, you don't need to make any changes, just make sure the dataset file format is appropriate..

kulkarnikeerti commented 2 years ago

@yjh0410 Okay, thanks a lot.