基于YOLOv7x的接触网吊弦缺陷检测方法

作者：王晓明; 陈智宇; 董文涛^*; 姚道金; 黄贻凤

来源：华东交通大学学报, 2024, 1-10.

DOI：10.16749/j.cnki.jecjtu.20240326.006

• 接触网吊弦缺陷检测
• YOLOv7x
• Swin Transformer
• SIoU损失函数
• Coordinate Attention

针对铁路运行时由于接触网吊弦缺陷造成的安全隐患问题，提出一种基于改进YOLOv7x的接触网吊弦缺陷识别方法。首先在主干特征提取层的末端引入Swin Transformer网络替换原有的扩展高效层聚合网络模块，提高网络掌握全局信息的能力；再引用SIoU(SCYLLA-IoU)损失函数替换原网络的损失函数，为预测框收敛过程添加方向惩罚机制；最后使用Coordinate Attention注意力机制融合颈部层中的扩展高效层聚合网络模块，增强颈部网络模块的全局感受野。实验仿真结果表明此改进算法训练出的模型精度达到95.9%，相较于原YOLOv7x算法检测精度提高了4.7%，检测速度也达到了52帧每秒；本改进算法解决了在吊弦缺陷识别领域检测效率低下的问题，在实际作业中能够提高接触网吊弦缺陷排查工作的效率。

• 单位
  华东交通大学