A Cross Entropy Based Deep Neural Network Model for Road Extraction from Satellite Images

被引:21
作者
Shan, Bowei [1 ]
Fang, Yong [1 ]
机构
[1] Changan Univ, Sch Informat Engn, Xian 710064, Peoples R China
关键词
cross entropy; encoder-decoder; road extraction; deep convolutional neural network;
D O I
10.3390/e22050535
中图分类号
O4 [物理学];
学科分类号
0702 ;
摘要
This paper proposes a deep convolutional neural network model with encoder-decoder architecture to extract road network from satellite images. We employ ResNet-18 and Atrous Spatial Pyramid Pooling technique to trade off between the extraction precision and running time. A modified cross entropy loss function is proposed to train our deep model. A PointRend algorithm is used to recover a smooth, clear and sharp road boundary. The augmentated DeepGlobe dataset is used to train our deep model and the asynchronous training method is applied to accelerate the training process. Five salellite images covering Xiaomu village are taken as input to evaluate our model. The proposed E-Road model has fewer number of parameters and shorter training time. The experiments show E-Road outperforms other state-of-the-art deep models with 5.84% to 59.09% improvement, and can give the accurate predictions for the images with complex environment.
引用
收藏
页数:16
相关论文
共 25 条
[1]  
Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
[2]  
[Anonymous], P INT C COMP VIS
[3]  
[Anonymous], 2017, ARXIV PREPRINT ARXIV
[4]  
[Anonymous], 2017, Computing Research Repository, DOI DOI 10.4271/2018-01-1635
[5]  
[Anonymous], 2016, PROC CVPR IEEE, DOI DOI 10.1109/CVPR.2016.492
[6]  
[Anonymous], 2018, P EUR C COMP VIS ECC
[7]  
[Anonymous], P CVPR BOST MA US 7
[8]  
[Anonymous], 2017, ARXIV160902907
[9]   Waterfall Atrous Spatial Pooling Architecture for Efficient Semantic Segmentation [J].
Artacho, Bruno ;
Savakis, Andreas .
SENSORS, 2019, 19 (24)
[10]   RoadTracer: Automatic Extraction of Road Networks from Aerial Images [J].
Bastani, Favyen ;
He, Songtao ;
Abbar, Sofiane ;
Alizadeh, Mohammad ;
Balakrishnan, Hari ;
Chawla, Sanjay ;
Madden, Sam ;
DeWitt, David .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :4720-4728