基于YOLOv5的矿井火灾视频图像智能识别方法

Intelligent identification method of mine fire video images based on YOLOv5

  • 摘要: 针对煤矿井下光照分布不均匀造成视频图像失真,火灾识别精度低等问题,提出了一种矿井火灾视频图像智能识别方法。该方法以YOLOv5为识别模型,采用K-means算法对传统的暗通道图像去雾算法进行改进,并用改进算法对采集的火焰图像进行去雾处理,提高矿井火灾视频图像识别精度;为减少静态背景对火灾识别的影响,采用帧差法与混合高斯模型融合算法,对动态演化的火焰图像进行特征提取,并采用形态学处理算法消除图像中存在的缺口,从而得到更加完整的火焰目标图像;对火灾视频图像数据集进行标注,并输入到YOLOv5算法模型进行训练及测试。结果表明:基于YOLOv5的矿井火灾视频图像智能识别方法平均精度为92%,损失函数为0.6,比传统算法Alexnet,VGG16,Inceptionv3的平均精度分别高9.6%,13.5%,4.9%,表明该方法检测速度快、精度高,可有效提高矿井火灾识别准确率。

     

    Abstract: In order to solve the problems of video image distortion caused by uneven light distribution and low accuracy of fire identification in coal mines, an intelligent identification method of mine fire video images is proposed. The method uses YOLOv5 as the identification model and uses K-means algorithm to improve the traditional dark channel image defogging algorithm to defog the collected flame images and improve the identification accuracy of mine fire video images. In order to reduce the impact of static background on fire identification, the fusion algorithm of frame difference method and Gaussian mixture model is used to extract the characteristics of the dynamically evolved flame images, and the morphological processing algorithm is used to eliminate the gaps in the images so as to obtain more complete flame target images. The fire video image data set is annotated and input to the YOLOv5 algorithm model for training and testing. The results show that the average accuracy of the intelligent identification method of mine fire video images based on YOLOv5 is 92% with a loss function of 0.6, which is 9.6%, 13.5% and 4.9% higher than that of the traditional algorithms, Alexnet, VGG16 and Inceptionv3 respectively, indicating that this method has fast detection speed and high accuracy, and can improve the accuracy of mine fire identification effectively.

     

/

返回文章
返回