Code for the 1st place solution in DeepGlobe Road Extraction Challenge.
- Cuda 8.0
- Python 2.7
- Pytorch 0.2.0
- cv2
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.
- Run
python train.py
to train the default D-LinkNet34.
- Run
python test.py
to predict on the default D-LinkNet34.