search

yijingru / BBAVectors-Oriented-Object-Detection

[WACV2021] Oriented Object Detection in Aerial Images with Box Boundary-Aware Vectors
MIT License
462 stars 87 forks source link

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

Open ayennam opened 2 years ago

ayennam commented 2 years ago

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

yijingru commented 2 years ago

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 https://github.com/yijingru/BBAVectors-Oriented-Object-Detection/blob/9ce1f3d250ec0625c53ae05f1b37b4ee3085e2a8/main.py#L34 and https://github.com/yijingru/BBAVectors-Oriented-Object-Detection/blob/9ce1f3d250ec0625c53ae05f1b37b4ee3085e2a8/train.py#L25

tcotte commented 1 year ago

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