Skip to content

Lizw14/DOTA_devkit

Repository files navigation

The code is useful for DOTA or ODAI. The code provide the following function

  • Load and visulize the data.
  • Evaluate the result.
  • Split and merge the picture and label.

What is DOTA?

Dota is a large-scale dataset for object detection in aerial images. It can be used to develop and evaluate object detectors in aerial images. We will continue to update DOTA, to grow in size and scope and to reflect evolving real-world conditions. Different from general object detectin dataset. Each instance of DOTA is labeled by an arbitrary (8 d.o.f.) quadrilateral. For the detail of DOTA-v1.0, you can refer to our paper.

What is ODAI?

ODAI is a contest of object detetion in aerial images on ICPR'2018. It is based on DOTA-v1. The contest is ongoing now.

Installation

  1. install swig
    sudo apt-get install swig
  1. create the c++ extension for python
    swig -c++ -python polyiou.i
    python setup.py build_ext --inplace

Usage

  1. For read and visualize data, you can use DOTA.py
  2. For evaluation the result, you can refer to the "dota_evaluation_task1.py" and "dota_evaluation_task2.py"
  3. For split the large image, you can refer to the "ImgSplit"
  4. For merge the results detected on the patches, you can refer to the ResultMerge.py

An example is shown in the demo. The subdirectory of "basepath"(which is used in "DOTA.py", "ImgSplit.py") is in the structure of

.
├── images
└── labelTxt

About

EE Thesis Code@THU: DOTA_devkit

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published