一种基于计算机视觉的滑环室碳刷打火智能检测方法

本发明公开一种基于计算机视觉的滑环室碳刷打火智能检测方法，它包括如下步骤：步骤1，制作集电环碳刷打火数据集；步骤2，构建基于注意力机制的改进YOLOv5s模型；步骤3，构建训练集、验证集和测试集；其中训练集用于训练神经网络，验证集用于调整模型超参数，测试集用于测试模型性能；步骤4，利用训练集训练构建好的改进YOLOv5s模型，再利用验证集调整模型的超参数后，将测试集输入到训练好的深度学习网络中进行性能评估；步骤5，保存模型并进行模型部署，使用训练好的改进YOLOv5s模型进行集电环碳刷打火检测，当检测到碳刷打火时，触发报警信号；本发明在出现碳刷打火的初期就可及时发现并发出早期告警，降低因碳刷打火带来的一系列影响和后果。