关键词:
动车组
TEDS
YOLOv5
Swin-Transformer
解耦头
故障检测
摘要:
针对动车组运行故障动态图像检测系统(Train of EMU Failures Detection System,TEDS)人工方式故障识别易漏检、强度大的问题,本文提出了一种改进的YOLOv5n算法的动车制动盘螺栓故障检测模型SSD-YOLO。针对制动盘螺栓缺损部位较小、缺损与正常样本特征相似的特点改进了YOLOv5n的网络结构。为了使模型能灵活适应不同尺度的特征,本文将主干网络中的下采样卷积层替换为可切换的空洞卷积,扩大感受野。为了提升网络获取全局信息和进行上下文信息交互的能力,本文在主干网络末端融合SwinTransformer模块。为了更好地处理不同尺度和分辨率的语义信息,本文将YOLO系列的耦合检测头替换为能够分别提取目标位置和类别信息的Efficient decoupled head解耦头。为了进一步提升算法的训练收敛速度,本文使用定位能力更好的SIoU替换CIoU损失函数。本文使用人工标注的动车组制动盘螺栓部位样本,对网络进行训练。实验证明,改进算法在动车组制动盘螺栓故障数据集上的检测m AP值较YOLOv5n原模型的91.5%提升了6.8%,达到98.3%,在RTX3090显卡上实现了89FPS的检测帧速率,是YOLOv5l的1.6倍,是YOLOX-S的3.7倍,满足TEDS故障检测的实时性要求。SSD-YOLO算法能够快速检测出动车组制动盘螺栓的缺失故障情况,减少分析员的人工作业量,为今后的动车组状态检修研究提供参考。