基于改进YOLOv5算法的接触网绝缘子定位方法 |
| |
引用本文: | 刘仕兵,周诗涵,但业光.基于改进YOLOv5算法的接触网绝缘子定位方法[J].华东交通大学学报,2024(1):105-112. |
| |
作者姓名: | 刘仕兵 周诗涵 但业光 |
| |
作者单位: | 华东交通大学电气工程学院 |
| |
摘 要: | 【目的】针对高速铁路接触网绝缘子在复杂背景下检测效率不高的问题提出一种检测算法。【方法】首先对样本数据集进行大规模扩充,在原有YOLOv5s算法的基础上,为有效的提升模型的表征力,增加ECA注意力机制,进行无降维的跨信道方式来聚焦绝缘子位置信息;使用BiFPN特征金字塔网络,进行多尺度的特征融合来丰富语义信息;选用Meta-ACON自适应控制激活函数,在函数允许的最大范围内,严格把控函数的上下限,防止模型出现失控现象;将原有GIOU损失函数更换为EIOU损失函数,从梯度的角度对锚框进行更深一步的划分,进而提升网络的收敛速度。【结果】实验结果表明,通过对YOLOv5s改进后的检测算法,可以对绝缘子进行更精确的定位与识别,准确率达到了99.4%。【结论】所提出的检测算法为绝缘子定位检测提供了更加准确快捷的方法。
|
关 键 词: | 绝缘子 风格迁移 YOLOv5s 注意力机制 双向融合特征网络 |
|
|