/DFDet

Offical Implementation for Oriented Object Detection via Contextual Dependence Mining and Penalty-Incentive Allocation

Primary LanguagePythonApache License 2.0Apache-2.0

โœจ๐““๐“•๐““๐“ฎ๐“ฝโœจ

Typing SVG

Introduction

This is the official implementation๐Ÿข of DFDet, which is implemented on OBBDetection

Update

  • (2023-05-20) Release DFDet๐Ÿ”ฅ.

Installation

Please refer to install.md for installation and dataset preparation.

Get Started

How to use OBBDetection

If you want to train or test a oriented model, please refer to oriented_model_starting.md.

How to Start DFDet

To help you start quickly, I prepare a simple bash script

Train

config=/path/to/config && work_dir=/path/to/work_dir && cuda=(device_id, like 0, 1, 2, 3 ...)
bash my_exps/train.sh ${config} ${work_dir} ${cuda}

Test

config=/path/to/config && ckpt=/path/to/checkpoint && save_dir=/path/to/results_save_dir && cuda=(same as above)
bash my_exps/test.sh ${config} ${ckpt} ${save_dir} ${cuda}

How to Deploy DFDet

TODO:

Cite

TODO:

License

This project is released under the Apache 2.0 license.