Chinese Space Science and Technology ›› 2022, Vol. 42 ›› Issue (3): 105-113.doi: 10.16708/j.cnki.1000-758X.2022.0041

Previous Articles     Next Articles

Improved U-Net network and its application of road extraction in remote sensing image

KONG Jiayuan ,ZHANG Hesheng   

  1. School of Mining Engineering, Taiyuan University of Technology, Taiyuan 030024,China 
  • Published:2022-06-25 Online:2022-06-22

Abstract: High resolution remote sensing image segmentation has a good application prospect in the military and civil fields, but due to the complex background conditions and the obstruction of interferences, the existing algorithms can’t extract road details from remote sensing images. Based on the improved U-Net network model, MDAU-Net (multi dimension attention U-Net) network structure model was proposed. The U-Net network structure was deepened to a seven-layer structure to improve the ability of fine segmentation of roads. A new multidimensional attention module,which was called MD-MECA (multi dimension modified efficient channel attention), was proposed to optimize the feature transfer in the coding part. DropBlock and Batch Normalization were used to resolve the overfitting during network training. The experimental results show that the improved algorithm can effectively improve the road extraction effect, and the accuracy rate on the test set reaches 97.04%.

Key words: remote sensing image, road extraction, U-Net network, multi-dimension attention, characteristics of the transfer