An Attention-Based Improved U-Net Neural Network Model for Semantic Segmentation of Moving Objects

被引:0
作者
Cui, Zhihao [1 ]
机构
[1] Pingdingshan Univ, Coll Phys Educ, Pingdingshan, Henan, Peoples R China
关键词
Attention mechanism; semantic segmentation; moving targets; computer vision;
D O I
10.1109/ACCESS.2024.3391249
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The precise semantic segmentation for moving targets has always been a challenging task in computer vision. The existing methods basically had some limitations in semantic segmentation, such as inability to handle deformation of moving targets, blurred boundaries, and other issues. To address the issue, this paper develops an improved U-Net model based on attention mechanism for this purpose. Firstly, we introduce an attention mechanism to enhance the perceptual ability of the U-Net model. By adding attention modules at different levels between the encoder and decoder, the network can pay more attention to the key features of moving targets at different levels. Then, we add a residual module to improve robustness and complete the capsule network for semantic segmentation of moving targets. By learning the deformation information of moving targets, the network can better adapt to moving targets with different shapes. We have conducted experimental verification on multiple public datasets. The experimental results show that the proposed method has superior performance in semantic segmentation tasks of moving targets. Compared with traditional U-Net-based models, the proposal shows significant improvements in accuracy and robustness.
引用
收藏
页码:57071 / 57081
页数:11
相关论文
共 38 条
[1]   DoubleU-NetPlus: a novel attention and context-guided dual U-Net with multi-scale residual feature fusion network for semantic segmentation of medical images [J].
Ahmed, Md. Rayhan ;
Ashrafi, Adnan Ferdous ;
Ahmed, Raihan Uddin ;
Shatabda, Swakkhar ;
Islam, A. K. M. Muzahidul ;
Islam, Salekul .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (19) :14379-14401
[2]   Multi parallel U-net encoder network for effective polyp image segmentation [J].
Al Jowair, Hamdan ;
Alsulaiman, Mansour ;
Muhammad, Ghulam .
IMAGE AND VISION COMPUTING, 2023, 137
[3]   Asymmetric dual-decoder-U-Net for pavement crack semantic segmentation [J].
Al-Huda, Zaid ;
Peng, Bo ;
Algburi, Riyadh Nazar Ali ;
Al-antari, Mugahed A. ;
AL-Jarazi, Rabea ;
Al-maqtari, Omar ;
Zhai, Donghai .
AUTOMATION IN CONSTRUCTION, 2023, 156
[4]   Automated skin lesion segmentation using attention-based deep convolutional neural network [J].
Arora, Ridhi ;
Raman, Balasubramanian ;
Nayyar, Kritagya ;
Awasthi, Ruchi .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 65
[5]   A Siamese Network Based U-Net for Change Detection in High Resolution Remote Sensing Images [J].
Chen, Tao ;
Lu, Zhiyuan ;
Yang, Yue ;
Zhang, Yuxiang ;
Du, Bo ;
Plaza, Antonio .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 :2357-2369
[6]  
Cordoba-Erazo M.F., 2015, Wireless and Microwave Technology Conference (WAMICON), 2015 IEEE 16th Annual, P1
[7]   The Cityscapes Dataset for Semantic Urban Scene Understanding [J].
Cordts, Marius ;
Omran, Mohamed ;
Ramos, Sebastian ;
Rehfeld, Timo ;
Enzweiler, Markus ;
Benenson, Rodrigo ;
Franke, Uwe ;
Roth, Stefan ;
Schiele, Bernt .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :3213-3223
[8]   Binary semantic segmentation for detection of prostate adenocarcinoma using an ensemble with attention and residual U-Net architectures [J].
Damkliang, Kasikrit ;
Thongsuksai, Paramee ;
Kayasut, Kanita ;
Wongsirichot, Thakerng ;
Jitsuwan, Chanwit ;
Boonpipat, Tarathep .
PEERJ COMPUTER SCIENCE, 2023, 9
[9]   Impact of attention mechanisms for organ segmentation in chest x-ray images over U-Net model [J].
de la Sotta, Tomas ;
Chang, Violeta ;
Pizarro, Benjamin ;
Henriquez, Hector ;
Alvear, Nicolas ;
Saavedra, Jose M. .
MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (16) :49261-49283
[10]   AAU-Net: Attention-Based Asymmetric U-Net for Subject-Sensitive Hashing of Remote Sensing Images [J].
Ding, Kaimeng ;
Chen, Shiping ;
Wang, Yu ;
Liu, Yueming ;
Zeng, Yue ;
Tian, Jin .
REMOTE SENSING, 2021, 13 (24)