• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science

Previous Articles     Next Articles

Road extraction algorithm based on
prediction and residual refinement networks
#br#  

XIONG Wei1,2,GUAN Lai-fu1,WANG Chuan-sheng1,TONG Lei1,LI Li-rong1,LIU Min1   

  1. (1.School of Electrical and Electronic Engineering,Hubei University of Technology,Wuhan 430068,China;
    2.Department of Computer Science and Engineering,University of South Carolina,Columbia,SC 29201,USA)
  • Received:2019-09-09 Revised:2019-11-30 Online:2020-04-25 Published:2020-04-25

Abstract:

Aiming at the problem of road detection in aerial images, a road extraction algorithm based on prediction and residual refinement networks is proposed. Firstly, the prediction network makes initial predictions. In order to improve the refinement ability of the segmentation network and learn higher- level road features, the dilated convolution and multi-kernel pooling modules are introduced in the prediction networks. Secondly, the residual refinement network will further refine the output of the prediction network and improve the ambiguity of the prediction network results. In addition, considering the small proportion of road pixels in aerial images, the network also combines binary cross entropy, structural similarity, and intersection over union loss functions to reduce road information loss. The experimental results on the Massachusetts road dataset show that the precision, recall, F value and accuracy reaches 99.3%, 95.7%, 97.3% and 95.1%, respectively. The intersection over union and structural similarity also reaches 94.8% and 84.3%, respectively. Compared with other algorithms, this proposed algorithm has certain application value.

 

 

 

Key words: aerial image, road extraction, deep learning, dilated convolution, multi-kernel pooling, loss function