Fully Convolutional Network for Automatic Road Extraction from Satellite Imagery

被引:119
作者
Buslaev, Alexander [1 ]
Seferbekov, Selim [2 ]
Iglovikov, Vladimir [3 ]
Shvets, Alexey [4 ]
机构
[1] Mapbox R&D Ctr, Minsk 220030, BELARUS
[2] Veeva Syst, D-60314 Frankfurt, Germany
[3] Lyft Inc, Engn Ctr Leve15, Palo Alto, CA 94304 USA
[4] MIT, 77 Massachusetts Ave, Cambridge, MA 02139 USA
来源
PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW) | 2018年
关键词
D O I
10.1109/CVPRW.2018.00035
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Analysis of high-resolution satellite images has been an important research topic for traffic management, city planning, and road monitoring. One of the problems here is automatic and precise road extraction. From an original image, it is difficult and computationally expensive to extract roads due to presences of other road-like features with straight edges. In this paper, we propose an approach for automatic road extraction based on a fully convolutional neural network of U-net family. This network consists of ResNet-34 pre-trained on ImageNet and decoder adapted from vanilla U-Net. Based on validation results, leader-board and our own experience this network shows superior results for the DEEPGLOBE - CVPR 2018 road extraction sub-challenge. Moreover, this network uses moderate memory that allows using just one GTX 1080 or 1080ti video cards to perform whole training and makes pretty fast predictions.
引用
收藏
页码:197 / 200
页数:4
相关论文
共 11 条
[1]  
[Anonymous], 2018, IEEE C COMP VIS PATT
[2]  
[Anonymous], 2018, IEEE GEOSCIENCE REMO
[3]   The PASCAL Visual Object Classes Challenge: A Retrospective [J].
Everingham, Mark ;
Eslami, S. M. Ali ;
Van Gool, Luc ;
Williams, Christopher K. I. ;
Winn, John ;
Zisserman, Andrew .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2015, 111 (01) :98-136
[4]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[5]  
Iglovikov V., 2018, TernausNet: U-Net with VGG11 encoder pre-trained on imagenet for image segmentation
[6]  
Iglovikov V., 2017, ARXIVLABS COMPUTER V
[7]  
Kingma D. P., P 3 INT C LEARN REPR
[8]  
Long J, 2015, PROC CVPR IEEE, P3431, DOI 10.1109/CVPR.2015.7298965
[9]   U-Net: Convolutional Networks for Biomedical Image Segmentation [J].
Ronneberger, Olaf ;
Fischer, Philipp ;
Brox, Thomas .
MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III, 2015, 9351 :234-241
[10]   A review of road extraction from remote sensing images [J].
Wang, Weixing ;
Yang, Nan ;
Zhang, Yi ;
Wang, Fengping ;
Cao, Ting ;
Eklund, Patrik .
JOURNAL OF TRAFFIC AND TRANSPORTATION ENGINEERING-ENGLISH EDITION, 2016, 3 (03) :271-282