Skip to content

wizard0822/object-detection-Aerial-

Repository files navigation

Object Detection on Aerial Imaging Dataset

Note: If you're interested in using it, feel free to ⭐️ the repo so we know!

Dataset

Satellite Imagery Multi-vehicles Dataset (SIMD). It comprises 5,000 images of resolution 1024 x 768 and collectively contains 45,303 objects in 15 different classes of vehicles including cars, trucks, buses, long vehicles, various types of aircrafts and boats. The source images are taken from public satellite imagery available in Google Earth and contain images of multiple locations from seven countries.

Data Annotations

With each image, the annotation is available as text file. The annotation format can be described as (c, xi, yi, w, h), where c is the object class starting from 0, (xi, yi) are the center of object and (w, h) are width and height respectively. All these values are percentages to the actual image.

Model Diagram

RetinaNet

YOLOv3

Faster-RCNN

Pre-Trained Models

The Pre-Trained models can be downloaded from google drive.

Installation

RetinaNet

 conda create -n Train python=3.7
 pip install -r code/keras-retinanet/requirements.txt

YOLOv3

 conda create -n ACV2YOLO python=3.6
 conda install -c anaconda tensorflow-gpu
 pip install Pillow==2.2.1
 pip instal numpy

Faster RCNN

 conda create -n FRCNN python=3.7
 conda install -c anaconda tensorflow-gpu
 pip install -r code/keras-frcnn/requirements.txt

Training

To start training run the following command:

 source train.sh

Testing

To start testing run the following command:

 source test.sh

Quantitatvie Results

Results Reported on Private Leaderboard of Challenge

Model MAP
RetinaNet 73

Results

Author

Maintainer Syed Nauyan Rashid ([email protected])

Acknowledgements

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published