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 #3

Closed Goubeast closed 2 years ago

Goubeast commented 2 years ago

Hi, How to train a custom dataset? My images don't require to be split like with DOTA.

braun-steven commented 2 years ago

Hey, you can have a look at the hrsc2016.py file and adapt the content according to your dataset.