一种基于图像识别的地铁列车电气箱盖的故障检测方法

摘要

本发明涉及列车故障检测技术领域，涉及一种基于图像识别的地铁列车电气箱盖的故障检测方法，包括：1、通过线扫相机获取电气箱盖部件的正负样本数据集；2、对数据集进行预处理，转换为伪彩色图；3、用正样本的数据集用于Nanodet的训练；4、通过裁剪得到分类正样本集；负样本上经过翻转，加入噪声进行数据增强，对负样本的数目进行扩充，定位后裁剪，得到分类负样本集；5、将分类负样本集与分类正样本集一起训练，最终得到故障检测模型；6、将待检测样本输入故障检测模型进行测试。本发明能够有效地检测出电气箱盖部件出现的故障，准确率高。