yijingru / BBAVectors-Oriented-Object-Detection

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

how to train on my own dataset? #8

Open JimmyLauren opened 4 years ago

JimmyLauren commented 4 years ago

Hi, thanks for ur excellent work. What modifications should i made to train on my own dataset?

yijingru commented 4 years ago

You can modify dataset_dota.py in datasets by replacing the following functions

    def load_img_ids(self):
        return None

    def load_image(self, index):
        return None

    def load_annoFolder(self, img_id):
        return None

    def load_annotation(self, index):
        return None

    def dec_evaluation(self, result_path):
        return None
JimmyLauren commented 4 years ago

thanks for reply. Another question, these code uses lib DOTA_devkit, but i could not find it, can you provide this file?

yijingru commented 4 years ago

It's this link: https://github.com/CAPTAIN-WHU/DOTA_devkit

JimmyLauren commented 4 years ago

thanks a lot