An improved U-Net method for the semantic segmentation of remote sensing images

被引:31
|
作者
Su, Zhongbin [1 ]
Li, Wei [1 ]
Ma, Zheng [1 ]
Gao, Rui [1 ]
机构
[1] Northeast Agr Univ, Coll Elect & Informat, Harbin 150030, Peoples R China
关键词
Deep learning; Artificial neural network; Remote sensing; U-Net; Semantic segmentation;
D O I
10.1007/s10489-021-02542-9
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Foremost deep neural network models trained in natural scenes cannot transfer and apply to remote sensing image semantic segmentation well. Studies have shown that fine-tuning methods containing model fusion can alleviate this dilemma. In this paper, we provide an approach used to improve U-Net and propose an end-to-end deep convolutional neural network (DCNN) combining the superiorities of DenseNet, U-Net, dilated convolution, and DeconvNet. We evaluated the proposed method and model on the Potsdam orthophoto data set. Compared with U-Net, our approach increases the PA, mPA, and mIoU evaluation indexes by 11.1%, 14.0%, and 13.5%, respectively; the segmentation speed increases by approximately 1.18 times and the number of parameters is 59.0% that of U-Net. The experiments demonstrate that for the semantic segmentation of high-resolution remote sensing images, using the combined dilated convolutions as the primary feature extractor, using the transposed convolution to restore the size of the feature maps, and reducing the number of layers is an effective method to improve the comprehensive performance of U-Net. This research enriches the models based on DCNNs and the modes of using DCNNs in a specific scene.
引用
收藏
页码:3276 / 3288
页数:13
相关论文
共 50 条
  • [1] An improved U-Net method for the semantic segmentation of remote sensing images
    Zhongbin Su
    Wei Li
    Zheng Ma
    Rui Gao
    [J]. Applied Intelligence, 2022, 52 : 3276 - 3288
  • [2] Improved U-Net remote sensing image semantic segmentation method
    Hu, Gongming
    Yang, Chuncheng
    Xu, Li
    Shang, Haibin
    Wang, Zefan
    Qin, Zhilong
    [J]. Cehui Xuebao/Acta Geodaetica et Cartographica Sinica, 2023, 52 (06): : 980 - 989
  • [3] Improved U-Net Network Segmentation Method for Remote Sensing Image
    Zhong, Letian
    Lin, Yong
    Sul, Yian
    Fang, Xianbao
    [J]. 2022 IEEE 6TH ADVANCED INFORMATION TECHNOLOGY, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (IAEAC), 2022, : 1034 - 1039
  • [4] Semantic Segmentation of High-Resolution Remote Sensing Images with Improved U-Net Based on Transfer Learning
    Zhang, Hua
    Jiang, Zhengang
    Zheng, Guoxun
    Yao, Xuekun
    [J]. INTERNATIONAL JOURNAL OF COMPUTATIONAL INTELLIGENCE SYSTEMS, 2023, 16 (01)
  • [5] Semantic Segmentation of High-Resolution Remote Sensing Images with Improved U-Net Based on Transfer Learning
    Hua Zhang
    Zhengang Jiang
    Guoxun Zheng
    Xuekun Yao
    [J]. International Journal of Computational Intelligence Systems, 16
  • [6] CTMU-Net: An Improved U-Net for Semantic Segmentation of Remote-Sensing Images Based on the Combined Attention Mechanism
    Li, Yuanjun
    Zhu, Zhiyu
    Li, Yuanjiang
    Zhang, Jinglin
    Li, Xi
    Shang, Shuyao
    Zhu, Dewen
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 : 10148 - 10161
  • [7] An Improved U-Net Method for Sequence Images Segmentation
    Wen, Peizhi
    Sun, Menglong
    Lei, Yongqing
    [J]. 2019 ELEVENTH INTERNATIONAL CONFERENCE ON ADVANCED COMPUTATIONAL INTELLIGENCE (ICACI 2019), 2019, : 184 - 189
  • [8] Semantic segmentation of remote sensing images combined with attention mechanism and feature enhancement U-Net
    Jiang, Jionghui
    Feng, Xi'an
    Ye, QiLei
    Hu, Zhongyi
    Gu, Zhiyang
    Huang, Hui
    [J]. INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (19) : 6219 - 6232
  • [9] U-Net Ensemble for Enhanced Semantic Segmentation in Remote Sensing Imagery
    Dimitrovski, Ivica
    Spasev, Vlatko
    Loshkovska, Suzana
    Kitanovski, Ivan
    [J]. REMOTE SENSING, 2024, 16 (12)
  • [10] Improved U-NET Semantic Segmentation Network
    Gao, Xueyan
    Fang, Lijin
    [J]. PROCEEDINGS OF THE 39TH CHINESE CONTROL CONFERENCE, 2020, : 7090 - 7095