Skip to content

miko7879/pytorch-remote-sensing

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Remote Sensing: Automatic Aircraft Detection in Satellite Imagery

Aircraft detection using the 'Airbus Aircraft Detection' dataset and Faster-RCNN with ResNet-50 backbone.

Original dataset can be found here: https://www.kaggle.com/airbusgeo/airbus-aircrafts-sample-dataset

To train the nextwork, first configure the ML parameters in the src/config.py file. Then run 'python engine.py' in the src directory. This will run for the configured number of epochs. You can see intermediate plots of training and validation loss in the output directory. Once trained, run 'python inference.py' in the src directory. This will draw bounding box predictions for all images in the data/extras directory and will write the images with predictions to the output/inference directory.

Analysis of results demonstrated a precision of greater than 0.99 and a recall of greater than 0.95, resulting in an F1 score of 0.97.

About

Aircraft detection using the 'Airbus Aircraft Detection' dataset and Faster-RCNN with ResNet-50 backbone.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages