- Pytorch Lightning
- https://www.dronedeploy.com/
Semantic segmentation implementation based on pytorch-lightning.
using main.py with predefined data paths (TODO: add hparams)
python main.py
Prediction example at
Inference.ipynb
file
Example results on Unet, after 16 epochs, trained on 768x768 resolution.
- ENet
- pspnet
- add hparams