一种基于改进YOLOv5算法的皮带托辊异常高温检测方法、系统、设备及介质

一种基于改进YOLOv5算法的皮带托辊异常高温检测方法、系统、设备及介质，包括：步骤1，设置皮带托辊的红外图像采集系统，利用红外图像采集系统采集皮带输煤过程中皮带托辊的红外热图，按照1：1的比例收集皮带托辊正常温度和异常高温的红外图像；步骤2，将步骤1收集到的红外图像进行人工标签标注，并划分为训练集和测试集；步骤3，将SE注意力机制添加到YOLOv5的主干网络中，建立改进的YOLOv5模型；步骤4，利用步骤2划分好的训练集和测试集对步骤3改进的YOLOv5模型进行训练，得到皮带托辊异常高温检测模型；系统、设备及介质用于实现该方法；本发明能够精确、高效的进行皮带输煤过程中托辊的高温故障诊断。