AHC-Net: a road crack segmentation network based on dual attention mechanism and multi-feature fusion

被引:5
作者
Shi, Lin [1 ]
Zhang, Ruijun [1 ]
Wu, Yafeng [1 ]
Cui, Dongyan [1 ]
Yuan, Na [2 ]
Liu, Jinyun [1 ]
Ji, Zhanlin [1 ,3 ]
机构
[1] North China Univ Sci & Technol, Hebei Key Lab Ind Intelligent Percept, Tangshan 063210, Peoples R China
[2] Tangshan Univ, Intelligence & Informat Engn Coll, Tangshan 063000, Peoples R China
[3] Zhejiang A&F Univ, Coll Math & Comp Sci, Hangzhou 311300, Peoples R China
关键词
Crack detection; Deep learning; U-Net; Multi-scale context fusion; Attention mechanism; UNET;
D O I
10.1007/s11760-024-03234-w
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
To solve the problem of incomplete and inaccurate pavement crack detection, an improved U-Net model based on dual attention mechanism and multi-feature fusion is proposed. Firstly, a new encoding module ACI is designed, which has the feature of multi-scale feature extraction, significantly improves the sensing ability of the damaged area, reduces the background interference, and realizes more accurate segmentation. Secondly, a new decoding module HAD is designed, which avoids the network degradation problem caused by gradient vanishing and the growth of network layers and can retain the most subtle feature information during the decoding process. Finally, convolutional block attention module (CBAM) is introduced in the encoding part to effectively extract global and local detail information, and the criss-cross attention mechanism is also introduced in the decoding part to prevent the loss of marginalized information. The model proposed in this article was tested on the public datasets DeepCrack, CrackSeg478, and AsphaltCrack300, and compared with other advanced methods. The experimental results indicate that this method can detect road cracks more accurately and possesses considerable robustness.
引用
收藏
页码:5311 / 5322
页数:12
相关论文
共 37 条
[1]  
Chen J., 2021, arXiv
[2]   U-Net-Based CNN Architecture for Road Crack Segmentation [J].
Di Benedetto, Alessandro ;
Fiani, Margherita ;
Gujski, Lucas Matias .
INFRASTRUCTURES, 2023, 8 (05)
[3]  
DrA UR, 2020, International Journal of Advanced Trends in Computer Science and Engineering, V9, P5393, DOI [10.30534/ijatcse/2020/175942020, 10.30534/ijatcse/2020/175942020, DOI 10.30534/IJATCSE/2020/175942020]
[4]  
Fan Z., 2018, ARXIV
[5]   MRA-UNet: balancing speed and accuracy in road crack segmentation network [J].
Gao, Xinwen ;
Tong, Bairui .
SIGNAL IMAGE AND VIDEO PROCESSING, 2023, 17 (05) :2093-2100
[6]   Deep learning based image recognition for crack and leakage defects of metro shield tunnel [J].
Huang, Hong-wei ;
Li, Qing-tong ;
Zhang, Dong-ming .
TUNNELLING AND UNDERGROUND SPACE TECHNOLOGY, 2018, 77 :166-176
[7]   CCNet: Criss-Cross Attention for Semantic Segmentation [J].
Huang, Zilong ;
Wang, Xinggang ;
Huang, Lichao ;
Huang, Chang ;
Wei, Yunchao ;
Liu, Wenyu .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :603-612
[8]   ACC-UNet: A Completely Convolutional UNet Model for the 2020s [J].
Ibtehaz, Nabil ;
Kihara, Daisuke .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT III, 2023, 14222 :692-702
[9]   Real-Time Polyp Detection, Localization and Segmentation in Colonoscopy Using Deep Learning [J].
Jha, Debesh ;
Ali, Sharib ;
Tomar, Nikhil Kumar ;
Johansen, Havard D. ;
Johansen, Dag ;
Rittscher, Jens ;
Riegler, Michael A. ;
Halvorsen, Pal .
IEEE ACCESS, 2021, 9 :40496-40510
[10]   MACU-Net for Semantic Segmentation of Fine-Resolution Remotely Sensed Images [J].
Li, Rui ;
Duan, Chenxi ;
Zheng, Shunyi ;
Zhang, Ce ;
Atkinson, Peter M. .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19