CVPR Workshops · DeepGlobe 2018

Stacked U-Nets with Multi-output for Road Extraction

Tao Sun, Zehui Chen, Wenxiang Yang, Yin Wang

In Procedings of 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW) 2018

Stacked U-Net architecture with auxiliary outputs, skip connections, and encoder-decoder blocks. — Satellite image with GPS trajectories

Abstract

With the recent advances of Convolutional Neural Networks(CNN) in computer vision, there have been rapid progresses in extracting roads and other features from satellite imagery for mapping and other purposes. In this paper, we propose a new method for road extraction using stacked U-Nets with multiple output. A hybrid loss function is used to address the problem of unbalanced classes of training data. Post-processing methods, including road map vectorization and shortest path search with hierarchical thresholds, help improve recall. The overall improvement of mean IoU compared to the vanilla VGG network is more than 20%.

BibTeX

@inproceedings{sun2018stacked,
  title={Stacked U-Nets with Multi-output for Road Extraction},
  author={Sun, Tao and Chen, Zehui and Yang, Wenxiang and Wang, Yin},
  booktitle={2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)},
  pages={187--1874},
  year={2018},
  organization={IEEE}
}