一种基于YOLOv5的高铁底部胶条故障检测方法及系统

摘要

本发明公开了一种基于YOLOv5的高铁底部胶条故障检测方法及系统，属于高铁底部胶条故障检测领域。步骤一、获取高铁中不同车厢的图像信息；步骤二、对图像信息中的高铁底部可能发生胶条故障的区域进行截取，得到子图像；步骤三、利用得到的子图像，构建训练样本集和测试样本集；步骤四、利用所构建的训练样本集和测试样本集，对YOLOv5模型进行训练以及测试，获得训练好的YOLOv5模型；步骤五、使用训练好的YOLOv5模型对待检测的高铁底部图像进行检测，输出检测结果。本发明与传统图像处理相比，在高铁底部胶条故障检测方面具有更高的准确率和泛化能力。