基于自注意力机制与卷积神经网络的隧道衬砌裂缝智能检测
铁道学报
页数: 11 2023-05-10
摘要: 卷积操作固有的局部性,导致难以捕获裂缝的全局特征表示。针对这一问题,以YOLOv5网络框架为基础,通过融合自注意力机制和卷积神经网络提出一种高精度隧道衬砌裂缝智能检测算法ST-YOLO。构建一种双分支特征提取模块,分支一采用基于自注意力机制的Swin Transformer网络提取裂缝的全局特征,分支二采用基于卷积神经网络的CSPDarkent提取裂缝的局部细节特征;采用卷积注... (共11页)