Awesome
DeepGlobe-Road-Extraction-Challenge
Code for the 1st place solution in DeepGlobe Road Extraction Challenge.
Requirements
- Cuda 8.0
- Python 2.7
- Pytorch 0.2.0
- cv2
Usage
Data
Place 'train', 'valid' and 'test' data folders in the 'dataset' folder.
Data is from DeepGlobe Road Extraction Challenge. You should sign in first to get the data.
Train
- Run
python train.py
to train the default D-LinkNet34.
Predict
- Run
python test.py
to predict on the default D-LinkNet34.