基于改进YOLOv3模型的接触轨表面缺陷检测方法

摘要：

为实现接触轨表面缺陷的高效检测,提出了一个基于改进YOLOv3模型的缺陷检测方法.一方面,采用轻量化的Ghost组件代替原始YOLOv3模型Darkent-53网络中的残差模块,来减少模型的尺寸和提高推理速度.另一方面,使用GIoU解决IoU在检测框和真值框不相交时损失函数无法反向传播的问题,并使用GIoU损失作为边界框损失加快模型收敛速度.利用在上海地铁16号线采集到的接触轨图像进行实验,精确率和召回率达到90.21％和89.64％,检测速度达到每张图片0.015 s,模型尺寸压缩到59.5 MB.

关键词：接触轨、缺陷检测、YOLOv3、Ghost组件

分类号：U216(铁路线路工程)

在线出版日期：2022-04-28（万方平台首次上网日期，不代表论文的发表时间）

页数：共3页

页码：30-32