基于YOLOv3-tiny的列车轮对踏面缺陷定位与测量方法

作者：金楷荣; 王俊平; 陈胜蓝

来源：控制与信息技术, 2022, (02): 69-75.

DOI：10.13889/j.issn.2096-5427.2022.02.200

• 踏面缺陷
• 目标检测
• YOLOv3-tiny
• 傅里叶变换
• 深度学习

针对基于深度学习的轮对踏面缺陷检测算法中检测框尺寸精度不足的问题，文章提出一种结合YOLOv3-tiny与传统图像算法的踏面缺陷检测算法，以实现低CPU消耗的列车轮对踏面缺陷快速定位与几何参数的精确测量。其首先对工业场景下获取到的小样本进行图像增强，然后采用YOLOv3-tiny算法对踏面缺陷进行迁移学习，实现缺陷的粗定位；为了解决检测框过大、过小问题，其利用傅里叶变换、带阻滤波器、阈值分割等传统图像算法构建缺陷尺寸测量模型，对粗定位出的缺陷进行轮廓提取并优化检测框尺寸，最后精确计算缺陷的位置和尺寸。缺陷定位实验结果显示，交并比（IoU）阈值为0.5时，缺陷识别的平均精度为89.4%,CPU消耗率不超过10%；缺陷测量实验结果表明，该算法能够对90个检测框中的74个进行优化，从而获得更加精确的缺陷尺寸，验证了本文检测算法在改善检测框尺寸精度问题上的有效性。

• 单位
  株洲中车时代电气股份有限公司