A PyTorch implementation of U-Net for aerial imagery semantic segmentation.
Read the article at Towards data science
UNet-AerialSegmentation
├── dataloader.py
├── losses.py
├── model.py
├── train.py
└── inference.py
!python train.py --num_epochs 2 --batch 2 --loss focalloss