Custom dataset with new categories

braun-steven / DAFNe

Code for our paper "DAFNe: A One-Stage Anchor-Free Deep Model for Oriented Object Detection".

MIT License

60 stars 12 forks source link

Custom dataset with new categories #6

Closed HappyBahman closed 2 years ago

HappyBahman commented 2 years ago

Hi, could you please guide me on how to change the code to work with a custom dataset with new labels (and different number of categories)? I did see #3 but I couldn't find anything out about new categories. Thanks.

HappyBahman commented 2 years ago

I solved this by changing the dota_evaluation.py file.

braun-steven commented 2 years ago

Glad to hear that! Sorry for the late reply. Yes, in addition to the data loading you need to adapt the evaluation procedure, as you've already found out :)