《计算机测量与控制杂志》发表论文赏析
作者:刘鹏杰,姚金杰,高晶,郭钰荣
单位:中北大学
摘要:针对道路病害检测中道路情况复杂、实时检测较为困难,缺检漏检等问题,采集并制作了多类型道路病害数据集R-CRACK,在YOLOv5s模型基础上,在Neck模块上使用轻量化卷积GSConv模块替换部分标准卷积构建轻量化网络颈部;在Head模块上利用SimSPPF对空间金字塔池化方式进行改进并应用轻量级上采样算子CARAFE,而后得到GSC-YOLO模型。将GSC-YOLO模型利用矩形推理、图像加权,标签平滑处理方式对数据集R-CRACK中训练集部分进行训练。模型训练后的结果表示,与YOLOv5s基础模型相比,GSC-YOLO参数量减少6.8%、计算量减少4.8%、mAP(@.5)上涨了9.2%。利用改进前后的网络模型分别对单一及复杂环境下的道路病害进行检测,通过对比不同模型的检测效果,证明了GSC-YOLO模型针对YOLOv5s缺检漏检等问题有所改进,此类轻量化检测网络对解决道路病害检测有着重要意义。
关键词:道路病害检测;YOLOv5;缺陷漏检;CARAFE;GSConv
基金资助:山西省重点研发计划项目(202102010101002);内蒙古自治区科技计划项目(2022YFSJ0031)