摘要
针对目前路面缺陷检测精度不高的问题,基于目前综合性能较强的YOLOv7 模型进行改进,提出了融合卷积注意力模块的CBAM-YOLOv7 模型.该模型通过重参数化、高效聚合、辅助训练等模块来加快模型的收敛过程,还通过融合卷积注意力机制来提升模型的检测精度.面向路面缺陷多分类问题,在公开数据集上分别使用 Faster R-CNN、YOLOv6、YOLOv7、CBAM-YOLOv7 模型进行实验验证分析,利用mAP值、F1 值、FPS值作为模型的精度与效率评价指标.实验结果显示:融合CBAM-YOLOv7 的路面缺陷检测结果的mAP值和F1 值分别能达到83.75%和67.8%,FPS值能达到51.22 Hz,相较于其他模型均有明显提高.
基金项目
湖北省交通厅科技项目(2020-186-1-6)
湖北省安全生产专项科技项目(KJZX202007010)