yijingru / BBAVectors-Oriented-Object-Detection

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

How to Train my own dataset which is in DOTA format?? #80

Open saik2121 opened 3 years ago

saik2121 commented 3 years ago

I have my trainval.txt ,test.txt files ready. How to train on these images.

yijingru commented 3 years ago

You may modify dataset_dota.py to adapt the code to your own data or put images the same as DOTA, such as

data_dir/
        images/*.png
        labelTxt/*.txt
        trainval.txt
        test.txt