choiszt / spade_dlinknet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

License:MIT License