一种基于图像识别的动车转向架部件缠绕异物检测方法

本发明公开了一种基于图像识别的动车转向架部件缠绕异物检测方法，属于铁路动车运行状态故障检测技术领域，该方法通过运用SimCLR模型训练特征提取器，以能够运用无标注的数据集实现对特征提取器的训练，再采用少量的有标注的数据集对目标检测器进行训练，以完成整个模型的训练过程，基于此训练好的模型将待检测图像输入至模型中能够实现动车转向架部件缠绕异物检测。本发明的方法所运用的训练集大部分为无标注的训练集，因此极大减少由大量的故障样本数据及标注文件产生的人力物力成本。