saiseg

This repository is for satellite/aerial image (sai) segementation & classification.

networks implemented

• U-net
• Segnet
• Linknet

next coming

• Stacked U-net
• [Fusenet]

usage

1. Download the ISPRS dataset
2. Modify the "data:path" in saiseg/configs/isprs_*.yml with your correct dataset path.
3. [sudo] python train.py

dataLoaders implemented

• ISPRS dataset

todo

1. To add the visualization
2. To add other datasets implemented

requirements

• numpy
• scipy
• pytorch >=0.3.1
• torchvision ==0.2.0
• sklearn

related works references

• Awesome Semantic Segmentation
• Related satellight/aerial image (sai) datasets
• CVPR2018 satellight imagery analysis papers
• DeepGlobe-Road-Extraction-Challenge

acknowlegements

This project is implemented with making references from @meetshah1995.