Current Issue Cover
结合Faster R-CNN的多类型火焰检测

回天,杜晗(新疆大学电气工程学院,新疆,乌鲁木齐,830047)

摘 要
目的 火焰检测可有效防止火灾的发生。针对目前火焰检测方法,传统图像处理技术的抗干扰能力低、泛化性不强,检测效果对数据波动比较敏感;机器学习方法需要根据不同的场景设定并提取合适火焰的特征,过程比较繁琐。为此提出一种基于Faster R-CNN的多类型火焰检测方法,避免复杂的人工特征提取工作,在面对复杂背景、光照强度变化和形态多样的火焰图像时依然保证较好的检测精度。方法 该方法是基于深度学习的思想,利用卷积神经网络自动学习获取图像特征。首先,利用自建数据集构建视觉任务。根据火焰的尖角特性、直观形态和烟雾量等,将火焰类数据划分为单尖角火焰、多尖角火焰和无规则火焰三类。此外,通过深度网络特征可视化实验发现,人造光源与火焰在轮廓上具有一定的相似性,为此建立了人造光源圆形和方形两个数据集作为干扰项来保证检测模型的稳定性;然后,细化训练参数并调整预训练的卷积神经网络结构,改动分类层以满足特定视觉任务。将经过深度卷积神经网络中卷积层和池化层层抽象的到的图像特征送入区域生成网络来进行回归计算,利用迁移学习的策略得到每一类目标物体相应的探测器;最后,得到与视觉任务相关的目标检测模型,保存权重和偏置参数。并联各类目标物体的子探测器作为整体探测器使用,检测时输出各类探测器的分数,得分最高的视为正确检测项。结果 首先,利用训练好的各探测器与相应测试集样本进行测试,然后,再利用各类目标物的测试集来测试其它类探测器的检测效果,以此证明各探测器之间的互异性。实验表明,各类探测器都具有较高的专一性,大大降低了误判的可能性,对于形变剧烈和复杂背景的火焰图像也具有良好的检测准确率。训练得到的检测模型在应对小目标、多目标、形态多样、复杂背景和光照变化等检测难度较大的情况时,均能获得很好的效果,测试集结果表明各类探测器的平均准确率提高了3.03%~8.78%不等。结论 本文提出的火焰检测方法,通过挖掘火焰的直观形态特征,细分火焰类别,再利用深度卷积神经网络代替手动特征设置和提取过程,结合自建数据集和根据视觉任务修改的网络模型训练得到了检测效果良好的多类型火焰检测模型。利用深度学习的思想,避免了繁琐的人工特征提取工作,在得到了较好的检测效果同时,也保证了模型具有较强的抗干扰能力。本文为解决火焰检测问题提供了更加泛化和简洁的解决思路。
关键词
Multi-type flame detection combined with Faster R-CNN

Hui Tian,Du Han(School of Electrical Engineering, Xinjiang University, Urumqi, Xinjiang,830001)

Abstract
Objective Flame detection can effectively prevent the occurrence of fire. For current flame detection methods, traditional image processing techniques have low anti-interference ability and generalization and the detection effect is more sensitive to data fluctuations; the machine learning method needs to set and extract the characteristics of a suitable flame according to different scenarios, and the process is complicated. In order to avoid complex artificial feature extraction and ensure good detection accuracy when face the complex backgrounds, lighting changes and various forms of flame images, a multi-type flame detection method based on Faster R-CNN is proposed.Method This method is based on the idea of deep learning and uses Convolutional Neural Network to automatically learn to acquire image features. First, build visual tasks by using self-built data sets. According to the sharp angle characteristics of the fire, the visual shape and the amount of smoke, the flame data is divided into three types: single point flames, multi points flames and shapeless flames. In addition, through the in-depth network feature visualization experiments, it is found that the artificial light source and the flame have similarities in contour, so that two data sets of artificial light sources (circular and square) were established as interference items to ensure the stability of the detection model; then, the training parameters are refined and the pre-trained Convolutional Neural Network structure is adjusted, and the classification layer is modified to meet specific visual tasks. The image features abstracted by the convolutional layers and the pooling layers in the deep Convolutional Neural Network are sent to the Region Proposal Network for regression calculation, and the corresponding detectors for each type of target object are obtained using the transfer learning strategy; finally, get the target detection model related to visual tasks, save the weights and bias parameters. The sub-detectors of various target objects in parallel are used as the overall detector and the scores of various detectors are output during the detection, so the highest score is regarded as the correct detection item.Result First, use the trained detectors and the corresponding test dataset to test, and then use the test sets of various targets to test the detection effect of other types of detectors, the mutuality between the detectors can be proved by this way. Experiments show that all kinds of detectors have high specificity, which greatly reduces the possibility of misjudgment. It also has good detection accuracy for flame images with sharp deformation and complex background. The detection model obtained through training can achieve good results when dealing with difficult situations such as small targets, multiple targets, various forms, complex backgrounds and lighting changes, the test set results show that the average accuracy of various types of detectors has increased between 3.03% and 11.42%.Conclusion The flame detection method proposed in this paper subdivides the flame category by excavating the visual morphological characteristics of the flame, then using the deep Convolutional Neural Network instead of manual feature setting and extraction process. Combining self-built data sets and network models modified according to visual tasks, a multi-type flame detection model with good detection results is obtained. Using the idea of deep learning, it avoids the tedious work of artificial feature extraction and has a good detection effect, and it also ensures that the model has strong anti-interference ability. This article provides a more general and concise solution to the problem of flame detection.
Keywords
QQ在线


订阅号|日报