基于改进YOLOv5的铁路接触网绝缘子检测方法

对铁路接触网绝缘子进行准确识别是实现绝缘子缺陷检测的关键前提。为解决铁路4C系统采集的夜间绝缘子图像存在的不同方向的纹理特征差异、图像明暗不均等问题,提出一种基于改进YOLOv5的铁路接触网绝缘子检测算法。通过采用循环曝光生成思想解决不均匀明暗问题,设计SREG模块,用于改善图像表面明暗不均的问题;在骨干模型中重新设计C3模块,融入旋转不变卷积,更好地提取绝缘子不同方向的纹理特征。为验证改进后模型的性能,在测试集上进行试验。结果表明,基于改进YOLOv5的铁路接触网绝缘子检测算法能适用于不同方向纹理的绝缘子识别,识别的平均精度达到99.3%,F1值为98.9%,可实现夜间4C系统下铁路接触网绝缘子的有效检测。