how to use my own dataset

[myasser63]

I want to use this model to train on my own dataset. If I use my dataset instead of COCO dataset with the same directory format will it work?

[uvipen]

Yes, definitely

[myasser63]

@uvipen Could ypu guide me on how to use my own data. I tried but I couldn't make work

[matt-guay-percipient]

@myasser63 you couldn't make it work because there are hard-coded dataset directory structure assumptions in the code. You'll have to modify how train.py handles the --data-path arg, and how annFile and root are constructed in src/dataset.py around line 21.

My recommendation would be to change the --data-path arg to --image-root-dir and --coco-file args, then stop making assumptions about annFile and root (and also give them more descriptive names) in src/dataset.py.