CM-Unet: A Novel Remote Sensing Image Segmentation Method Based on Improved U-Net

被引:8
作者
Cui, Mengtian [1 ]
Li, Kai [1 ]
Chen, Jianying [1 ]
Yu, Wei [1 ,2 ]
机构
[1] Southwest Minzu Univ, Sch Comp Sci & Engn, Chengdu 610041, Peoples R China
[2] Chengdu Univ Informat Technol, Sch Software Engn, Chengdu 610225, Peoples R China
关键词
Remote sensing image; channel attention; multi-feature fusion; improved sub-pixel convolution; semantic segmentation;
D O I
10.1109/ACCESS.2023.3282778
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Semantic segmentation is an active research area for high-resolution (HR) remote sensing image processing. Most existing algorithms are better at segmenting different features. However, for complex scenes, many algorithms have insufficient segmentation accuracy. In this study, we propose a new method CM-Unet based on the U-Net framework to address the problems of holes, omissions, and fuzzy edge segmentation. First, we add the channel attention mechanism in the encoding network and the residual module to transmit information. Second, a multi-feature fusion mechanism is proposed in the decoding network, and an improved sub-pixel convolution method replaces the traditional upsampling operation. We conducted simulation experiments on the Potsdam, Vaihingen and GID datasets. The experimental results show that the proposed CM-Unet required segmentation time is approximately 62 ms/piece, the MIoU is 90.4% and the floating point operations (FLOPs) is 20.95 MFLOPs. Compared with U-Net, CM-Unet only increased the total number of parameters and floating point operations slightly, but achieved the best segmentation effect compared with the other models. CM-Unet can segment remote sensing images efficiently and accurately owing to its lower time consumption and space requirements; the precision of the segmentation results is better than other methods.
引用
收藏
页码:56994 / 57005
页数:12
相关论文
共 35 条
[1]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[2]   Region Merging Techniques Using Information Theory Statistical Measures [J].
Calderero, Felipe ;
Marques, Ferran .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2010, 19 (06) :1567-1586
[3]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851
[4]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[5]   A Modified KNN Method for Mapping the Leaf Area Index in Arid and Semi-Arid Areas of China [J].
Jiang, Fugen ;
Smith, Andrew R. ;
Kutia, Mykola ;
Wang, Guangxing ;
Liu, Hua ;
Sun, Hua .
REMOTE SENSING, 2020, 12 (11)
[6]   MM-Pyramid: Multimodal Pyramid Attentional Network for Audio-Visual Event Localization and Video Parsing [J].
Yu, Jiashuo ;
Cheng, Ying ;
Zhao, Rui-Wei ;
Feng, Rui ;
Zhang, Yuejie .
PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2022, 2022, :6241-6249
[7]   An attention-based U-Net for detecting deforestation within satellite sensor imagery [J].
John, David ;
Zhang, Ce .
INTERNATIONAL JOURNAL OF APPLIED EARTH OBSERVATION AND GEOINFORMATION, 2022, 107
[8]   Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1026-1034
[9]   Automated Semantic Segmentation for Autonomous Railway Vehicles br [J].
Katar, Oguzhan ;
Duman, Erkan .
TEHNICKI GLASNIK-TECHNICAL JOURNAL, 2022, 16 (04) :484-490
[10]   Truncation Cross Entropy Loss for Remote Sensing Image Captioning [J].
Li, Xuelong ;
Zhang, Xueting ;
Huang, Wei ;
Wang, Qi .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (06) :5246-5257