how to train my custom dataset is in coco json format?

[ayennam]

here using virat drone dataset, is in coco json format how to train with it, @yijingru

[yijingru]

Hi, for a custom dataset, you may need to make a new datasets/dataset_custom.py to load images and annotations. You may find datasets/dataset_dota.py as an example. Then add your custom dataset to

BBAVectors-Oriented-Object-Detection/main.py

Line 34 in 9ce1f3d

dataset = {'dota': DOTA, 'hrsc': HRSC}

and

BBAVectors-Oriented-Object-Detection/train.py

Line 25 in 9ce1f3d

self.dataset_phase = {'dota': ['train'],

[tcotte]

You can transform your COCO json format dataset into DOTA format using this repo :
https://github.com/tcotte/coco2yolo_obb