choiszt / spade_dlinknet

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.

Download trained D-LinkNet34

• Dropbox
• 百度网盘