基于残差与注意力机制的道路裂缝检测U-Net改进模型 |
| |
引用本文: | 于海洋,景鹏,张文涛,谢赛飞,滑志华,宋草原.基于残差与注意力机制的道路裂缝检测U-Net改进模型[J].计算机工程,2023(6):265-273. |
| |
作者姓名: | 于海洋 景鹏 张文涛 谢赛飞 滑志华 宋草原 |
| |
作者单位: | 1. 河南理工大学自然资源部矿山时空信息与生态修复重点实验室;2. 河南交通发展研究院有限公司 |
| |
基金项目: | 国家自然科学基金(U1304402); |
| |
摘 要: | 道路裂缝是道路安全检测的重要部分,随着深度学习和计算机视觉的发展,利用深度学习对道路图像中裂缝信息提取的方法趋于成熟。现有深度学习道路裂缝检测方法对细小裂缝提取不完整以及受背景因素干扰,导致检测精度降低。基于CBAM注意力机制和残差网络,改进U-Net神经网络模型,构建一种融合残差和注意力机制的道路裂缝检测深度学习网络模型。该模型在U-Net网络的上采样和下采样过程中分别嵌入通道注意力机制和空间注意力机制。CBAM注意力机制在通道和空间维度上同时进行全局平均和全局最大混合池化,以提取更多有效的全局和局部细节信息。同时,在U-Net网络中融合残差模块,有效解决网络梯度消失、梯度爆炸以及网络退化的问题,进一步提高道路裂缝的检测能力。实验结果表明,在上采样和下采样过程中嵌入CBAM注意力机制网络的F1值提升到81.02%,相比U-Net原始网络,提升13.76个百分点。融合残差模块并在下采样过程中嵌入CBAM注意力机制网络的F1值达到85.82%,相比只嵌入CBAM注意力机制的网络,提升了4.8个百分点。
|
关 键 词: | 裂缝检测 深度学习 U-Net神经网络 注意力机制 残差结构 |
|
|