复杂背景环境下基于SSD_MobileNet深度学习模型的火焰图像识别研究

Flame recognition in complex environment based on deep learning model SSD_MobileNet

针对现有的火灾火焰图像识别方法在光照和红花等类似火焰干扰的复杂环境下存在错检和漏检的问题,提出一种基于SSD_MobileNet的复杂环境火焰区域标记方法。首先,将深度卷积神经网络SSD300的基础卷积网络VGG16替换为MobileNet网络,应用深度可分离卷积,降低网络参数,进而构建一种火焰图像检测的SSD_MobileNet模型;然后,迁移第一次训练模型所有的卷积层参数,初始化新的待训练模型;最后,加入新的数据样本用于削弱光照、红花等干扰对象的影响。通过与SSD300、以及深度学习的目标检测算法Faster R-CNN和YOLOv3-tiny对比,实验结果表明,构建的火焰检测和火焰区域标记SSD_MobileNet模型的综合性能优于Faster R-CNN和YOLOv3-tiny模型,更适用于实时火焰检测领域。

Aiming at the problem of false detection and missed detection of existing fire image recognition methods in the complex environment of light,red flower and other Flame-like interference,this paper proposes a method based on SSD_MobileNet for detecting the flame region in the complex environment.First,the basic convolutional network VGG16 of SSD300 was replaced by the MobileNet network using depthwise separable convolutions to reduce network parameters.A novel flame region detection model based on the improved SSD_MobileNet was constructed.All the convolution layer parameters of the first training model were then migrated to initialize the new model to be trained.Finally,the new data samples were added to reduce the effects of disturbing objects such as light and red flower.The experimental results showed that the comprehensive performance of the flame region detection model based on SSD_MobileNet constructed in this paper is better than other algorithms including SSD300,Faster R-CNN and YOLOv3-tiny,and it is more suitable for real-time flame detection.