计算机工程

作者：刘仕兵, 周诗涵

单位：华东交通大学电气工程学院, 江西 南昌 330013

关键词：旋转目标检测,YOLOv5网络结构,绝缘子,对齐卷积,注意力机制

基金：轨道交通基础设施性能监测与保障国家重点实验室开放课题(GJJ210652)。

针对已有高速铁路接触网绝缘子目标检测算法通常存在检测精度不高且忽视了绝缘子方向的问题,为了能更好地满足智能化巡检需求,提出一种基于改进YOLOv5的绝缘子旋转目标检测算法。首先,引入协调注意力(CA)和十字交叉注意力机制,高效提取绝缘子的有效特征及位置信息,同时利用骨干网络RepVGG架构,有效提升模型表征力和检测速度;在检测头的骨干网络中,采用对齐卷积(AC)模块解决了绝缘子目标的倾斜和特征不对齐问题,进一步调整了预测框与实际目标的对齐程度;最后,采用旋转完全交并比(R-CIoU)计算旋转损失函数,可以更好地实现对预测框的精准定位。实验结果表明,该算法可以实现对绝缘子不同方向的检测,在提升检测速度的同时平均精度均值(mAP)达到97.5%,能更好地满足绝缘子目标检测的需求。

来源：2024年第5期

《计算机工程》期刊编辑部

查看计算机工程杂志2024年第5期