列车故障识别方法及系统

列车故障识别方法及系统，涉及一种图像识别方法及系统。目的是为了克服现有列车故障识别方法中，由于偶发故障在数据集中占比较小，使得训练好的神经网络模型在识别列车偶发故障时的准确性识别较低，无法及时发现偶发故障的问题。其中方法，步骤如下：步骤一、采集列车待识别故障部位图像；步骤二、将故障部位图像输入至列车故障识别模型；列车故障识别模型包括改进监督对比学习网络；列车故障识别模型训练过程中，改进监督对比学习网络对数据集样本进行采样时，大数量数据集样本的采样概率和小数量数据集样本的采样概率都向1/K收敛；K为标签的数量；步骤三：若故障部位图像中存在故障，则列车故障识别模型报警。