计算机科学 ›› 2024, Vol. 51 ›› Issue (11): 198-204.doi: 10.11896/jsjkx.240100082
夏淑芳, 袁彬, 瞿中
XIA Shufang, YUAN Bin, QU Zhong
摘要: 自动化裂缝检测是确保混凝土路面品质并提升道路养护效率的关键。针对现有方法在关注裂缝特征方面的不足以及深层特征图中裂缝细节信息易丢失的问题,文中提出一种融合注意力机制与深层特征优化策略的网络模型。该模型以VGG-16作为主干网络,首先,在主干网络的中高层卷积后引入一种轻量级的置换注意力机制,旨在提高网络对裂缝特征的敏感性;其次,为了进一步增强对裂缝特征的捕捉能力,在每个阶段的侧边输出中嵌入相应的注意力模块;最后,提出一种空间可分离金字塔模块并设计了一种注意力融合模块,用以优化深层特征图,还原更多的裂缝细节。侧网络通过在多个层次上融合低层和高层特征,辅助生成最终的预测图像。该网络采用二分类交叉熵损失函数作为评价函数,经过训练的网络模型能够在复杂背景下准确地从输入的原始图像中识别裂缝位置。为验证所提方法的有效性,在DeepCrack,CFD和Crack500这3个公开数据集上将其与6种方法进行了比较,所提算法表现出卓越的性能,F-score值达到了87.19%。
中图分类号:
[1] ZHANG Q L,YANG Y B.SA-Net:Shuffle Attention for Deep Convolutional Neural Networks[C]//IEEE International Conference on Acoustics,Speech and Signal Processing.Toronto:IEEE,2021:2235-2239. [2] WU Y,YANG W,PAN J,et al.Asphalt Pavement Crack Detection Based on Multi-scale Full Convolutional Network[J].Journal of Intelligent & Fuzzy Systems,2021,40(1):1495-1508. [3] MA D,FANG H,WANG N,et al.A Real-time Crack Detection Algorithm for Pavement Based on CNN with Multiple Feature Layers[J].Road Materials and Pavement Design,2022,23(9):2115-2131. [4] ZHOU Q,QU Z,WANG S Y,et al.A Method of PotentiallyPromising Network for Crack Detection with Enhanced Convolution and Dynamic Feature Fusion[J].IEEE Transactions on Intelligent Transportation Systems,2022,23(10):18736-18745. [5] XU C,ZHANG Q,MEI L,et al.Dense Multiscale FeatureLearning Transformer Embedding Cross-Shaped Attention for Road Damage Detection[J].Electronics,2023,12(4):898. [6] ZHOU Q,QU Z,LI Y X,et al.Tunnel Crack Detection with Linear Seam Based on Mixed Attention and Multiscale Feature Fusion[J].IEEE Transactionson Instrumentation and Measurement,2022,71(1):1-11. [7] SU H,WANG X,HAN T,et al.Research on a U-Net BridgeCrack Identification and Feature-Calculation Methods Based on a CBAM Attention Mechanism[J].Buildings,2022,12(10):1561-1572. [8] SUN X,XIE Y,JIANG L,et al.Dma-net:Deeplab with Multi-scale Attention for Pavement Crack Segmentation[J].IEEE Transactions on Intelligent Transportation Systems,2022,23(10):18392-18403. [9] HANG J,WU Y,LI Y,et al.A Deep Learning Semantic Seg-mentation Network with Attention Mechanism for Concrete Crack Detection[J].Structural Health Monitoring,2023,22(5):3006-3026. [10] YANG L,BAI S,LIU Y,et al.Multi-scale Triple-attention Network for Pixelwise Crack Segmentation[J].Automation in Construction,2023,150:104853. [11] ZOU Q,ZHANG Z,LI Q,et al.Deepcrack:Learning Hierarchical Convolutional Features for Crack Detection[J].IEEE Transactions on Image Processing,2019,28(3):1498-1512. [12] LIU Y,YAO J,LU X,et al.DeepCrack:A Deep Hierarchical Feature Learning Architecture for Crack Segmentation[J].Neurocomputing,2019,338(21):139-153. [13] QU Z,WANG C Y,WANG S Y,et al.A Method of HierarchicalFeature Fusion and Connected Attention Architecture for Pavement Crack Detection[J].IEEE Transactions on Intelligent Transportation Systems,2022,23(9):16038-16047. [14] XU S,HAO M,LIU G,et al.Concrete Crack SegmentationBased on Convolution-deconvolution Feature Fusion with Holistically Nested Networks[J].Structural Control and Health Monitoring,2022,29(8):2965-2976. [15] FU J,LIU J,TIAN H,et al.Dual Attention Network for Scene Segmentation[C]//IEEE Computer Vision and Pattern Recognition.Long Beach:IEEE,2019:3146-3154. [16] SHI Y,CUI L,QI Z,et al.Automatic Road Crack Detection using Random Structured Forests[J].IEEE Transactions Intelligence Transport System,2016,17(12):3434-3445. [17] YANG F,ZHANG L,YU S J,et al.Feature Pyramid and Hierarchical Boosting Network for Pavement Crack Detection[J].IEEE Transactions on Intelligent Transportation Systems,2020,21(4):1525-1535. [18] CHEN J,LU Y,YU Q,et al.Transunet:Transformers Make Strong Encoders for Medical Image Segmentation[J].arXiv:2102.04306,2021. [19] LIU H,MIAO X,MERTZ C,et al.CrackFormer:Transformer Network for Fine-Grained Crack Detection[C]//IEEE International Conference on Computer Vision.IEEE,2021:3783-3792. [20] HAN C,MA T,HUYAN J,et al.CrackW-Net:A Novel Pavement Crack Image Segmentation Convolutional Neural Network[J].IEEE Transactions on Intelligent Transportation Systems,2021,23(11):22135-22144. [21] CHEN H,LIN H.An Effective Hybrid Atrous ConvolutionalNetwork for Pixel-level Crack Detection[J].IEEE Transactions on Instrumentation and Measurement,2021,70:1-12. |
|