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

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

扫码查看
[目的]针对铁路运行时由于接触网吊弦缺陷造成的安全隐患问题,提出一种基于改进YOLOv7x的接触网吊弦缺陷识别方法.[方法]首先在主干特征提取层的末端引入Swin Transformer网络替换原有的扩展高效层聚合网络模块,提高网络掌握全局信息的能力,再引用SIoU(SCYLLA-IoU)损失函数替换原网络的损失函数,为预测框收敛过程添加方向惩罚机制,最后使用coordinate attention(CA)注意力机制融合颈部层中的扩展高效层聚合网络模块,增强颈部网络模块的全局感受野.[结果]仿真结果表明,此改进算法训练出的模型精度达到95.9%,相较于原YOLOv7x算法检测精度提高了4.7%,检测速度也达到了52 帧/s.[结论]改进算法解决了在吊弦缺陷识别领域检测效率低下的问题,在实际作业中能够提高接触网吊弦缺陷排查工作的效率.
Detection Method of Catenary Hanging String Based on YOLOv7x
[Objective]Aiming at the potential risk caused by overhead contact wire defects during railway opera-tion,an improved YOLOv7x method for overhead contact wire defect identification is proposed.[Method]Firstly,Swin Transformer network is introduced at the end of the backbone feature extraction layer to replace the origi-nal extended and efficient layer aggregation network module,so as to improve the ability of the network to grasp global information.Then the SIoU(SCYLLA-IoU)loss function is used to replace the original network loss func-tion,and the direction penalty mechanism is added to the convergence process of the prediction frame.Finally,CA is integrated with the extended and efficient layer aggregation network module to enhance the global recep-tive field of the neck network module.[Result]Experimental simulation results show that the accuracy of the model trained with the improved algorithm reaches 95.9%,which is 4.7%higher than that of the original YO-LOv7x algorithm,and the detection speed reaches 52 frames per second.[Conclusion]The improved algorithm solves the problem of low detection efficiency in hanging strings defect identification,which may improve the ef-ficiency of detection of hanging strings defect in practice.

detection of catenary hanging string defectsYOLOv7xSwin TransformerSIoU loss functionCA

王晓明、陈智宇、董文涛、姚道金、黄贻凤

展开 >

华东交通大学电气与自动化工程学院,江西南昌 330013

华东交通大学土木建筑学院,江西南昌 330013

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

国家自然科学基金项目江西省教育厅科技项目

52165069GJJ2200621

2024

华东交通大学学报
华东交通大学

华东交通大学学报

CSTPCD
影响因子:0.748
ISSN:1005-0523
年,卷(期):2024.41(3)