基于Transformer的端到端路面裂缝检测方法

• 作者: 刘军 王慧民 张兴忠 张婷 郭美青
• 来源期刊: 太原理工大学学报
• 关键词: 路面裂缝检测 多尺度特征融合 Pre-LN Transformer网络 联合回归损失 端到端
• 年，卷（期）: 2022,（6）
• 所属期刊栏目: 信息与计算机
• 页码范围: 1143-1151
• 页数: 8页
• 分类号: TP391
• 语种: 中文
• DOI: 10.16355/j.cnki.issn1007-9432tyut.2022.06.021

摘要

针对路面裂缝检测场景中裂缝形状不规则和背景复杂引起的检测精度较低的问题，提出了一种基于transformer的端到端路面裂缝检测方法CrackFormerNet.首先，在特征提取阶段，引入多尺度特征融合机制，设计了Multi-Scale Transformer骨干网络，将不同下采样倍率特征图融合，提取细节信息丰富的裂缝纹理特征。其次，提出基于CIoU Loss和L1 Loss的联合回归损失函数来度量预测框和标签间距离，精确评估预测框检测效果。同时，为应对transformer模型收敛缓慢的问题，在编码器-解码器阶段使用Pre-LN Transformer结构，在残差连接内部使用层归一化，加速模型收敛。实验结果表明，方法MAP达到84.2%,优于主流基准方法。与DETR检测方法相比，模型收敛轮次压缩18.4%,检测精度提升3.6%,证明了该方法的有效性。