Semantic Segmentation - Drone Deploy Aerial Dataset

• Pytorch Lightning
• https://www.dronedeploy.com/

Semantic segmentation implementation based on pytorch-lightning.

Training

using main.py with predefined data paths (TODO: add hparams)

python main.py

Prediction

Prediction example at

Inference.ipynb 

file

Results

Example results on Unet, after 16 epochs, trained on 768x768 resolution.

Todo's

• ENet
• pspnet
• add hparams