基于多尺度特征融合的火灾检测模型

张坚鑫， 郭四稳， 张国兰， 谭 琳

(广州大学 计算机科学与网络工程学院，广东 广州 510006)

0 引言

火灾是全球范围内的灾难性破坏事件，会在短时间内对人民的生命和财产安全造成重大的损害。随着人工智能技术的不断发展，深度卷积神经网络已经在图像识别和检测方面展示了最先进的性能[1]。图像分类识别任务是对图像是否有该物体进行分类识别，图像检测任务是对物体的对应区域进行类别的标记。黄文锋等[2]将深度卷积神经网络应用于视频监控中火灾烟雾和火焰的识别和检测。Frizzi等[3]提出了用于视频火焰和烟雾识别的卷积神经网络，结果表明，基于卷积神经网络的方法比一些常规的视频火灾探测方法具有更好的性能。Sharma等[4]探讨了将Vgg16[5]和Resnet50[6]用于火灾识别。林作永等[7]使用Inception[8]、ResNet[6]、MobileNet[9]作为特征提取器Backbone，使用Faster R-CNN[10]、SSD[11]、R-FCN[12]作为深度学习检测框架进行火灾烟雾检测，结果发现，使用Inception V2[13]特征提取网络能增加检测精度，Faster R-CNN检测效果最好，SSD速度最快，但定位精度不够，R-FCN获得了速度与精度的平衡[14]。基于这些考虑，本文对双阶段目标检测网络Faster R-CNN在火灾图像的应用进行改进和研究，以实现对火灾区域的精准定位。

1 相关工作

在至今为止的目标检测算法中，基于深度学习的单阶段和双阶段目标检测算法都能够实现较高的检测准确率，但是两种算法各有优势：双阶段的精度更准，单阶段的速度更快。双阶段的Faster R-CNN系列方案第1步是先训练好特征提取网络；第2步是使用该网络生成很多候选框来检测目标。单阶段方案是将特征提取网络和生成候选框在同一个网络里完成。这里选用双阶段Faster R-CNN的目标检测方案进行改进。

在传统的目标检测任务中一般使用特征提取和机器学习结合的算法，特征提取方法往往有3种：第1种是最基本的，一般是颜色、纹理等底层的特征；第2种是中层特征，一般是经过特征挖掘学习之后得到的特征，包括PCA特征和LDA学习得到的特征；第3种是高层的特征，是将前面2种特征结合进一步挖掘计算得到的特征。基于这个考虑，双阶段Faster R-CNN目标检测任务中的特征提取网络能够融合多种尺度的特征来提升网络的效果。特征金字塔结构FPN方法[15]是对不同层中的信息进行融合，浅层网络包含更多的细节信息，高层网络则包含更多的语义信息。将卷积网络中浅层和高层的特征图进行累加，使其不会丢失过多信息，从而提升检测的效果。Inception系列的方法[16-17]就是通过增加网络的宽度来提升卷积神经网络的性能。不同大小的卷积核会提取到不同的信息：较大的卷积核会提取到图像中的全局性的特征图信息；较小的卷积核会提取到图像中局部的特征图信息。视觉注意力机制主要是模仿人的大脑信号处理机制，获取全局图像中需要重点关注的目标区域的细节信息。使用通道注意力机制和像素注意力机制可以将前景和背景之间的细节信息进行区分，从而增强检测的效果。

2 多尺度特征融合网络

本文引入了一种改进的Faster R-CNN检测火灾区域的目标检测框架，如图1所示。该方法主要是基于Resnet101网络进行特征提取，得到生成的每个候选框的类别分数，从而找出最佳结果。为充分利用火灾场景的火焰特征，使用了预训练效果较好的Resnet101作为特征提取网络，加入特征金字塔结构FPN，对浅层和高层特征进行提取，使用Inception Module结构提取多种卷积特征，并加入像素注意力机制和信道注意力机制，通过这些操作来突出特征使得生成候选框操作更加精准，以提高火灾区域定位的性能。

图1 多尺度特征融合网络结构Figure 1 Multi-scale feature fusion network structure

2.1 基础网络结构

一般认为浅层特征映射可以保留小对象的位置信息，而高层特征映射可以包含更高级别的语义提示。因此将语义信息充分的高层特征映射回分辨率较高、细节信息充分的底层特征，将两者以合适的方式融合来提升检测的效果。输入层的特征图尺寸为600×600，使用特征提取网络Resnet101的网络架构[5]。在Resnet101网络基础上采用了特征金字塔结构FPN，将Resnet101网络的卷积层2输出特征图和卷积层3输出特征图进行融合，如图1所示。通过这样的操作可以平衡语义信息和位置信息，同时忽略其他不太相关的特征。

2.2 Inception Module网络结构

将Resnet101中的卷积层2输出特征图输入到Inception Module结构中可以抽取更多的特征。Inception Module架构如图2所示。采用多个1×1卷积的作用是在相同尺寸的感受野中叠加更多的卷积，能提取到更丰富的特征，并且降低了计算复杂度。在上一层网络之后首先使用1×1卷积再进入较大的二维卷积的作用是当某个卷积层输入的特征数较多，对这个输入进行卷积运算将产生巨大的计算量；如果对输入先进行降维，减少特征数后再做卷积计算就会显著减少计算量。在这一层网络中还将较大的二维卷积分解成较小的一维卷积，例如将3×3卷积分解成1×3卷积和3×1卷积，将5×5卷积分解1×5卷积和5×1卷积，将7×7卷积分解成1×7卷积和7×1卷积。这个操作一方面可以减少参数、减轻过拟合，另一方面则增加了非线性扩展模型表达能力。非对称的卷积结构分解效果比对称地分解为几个相同的小卷积核效果更好，可以处理更丰富的空间特征，从而增加特征提取的多样性。

图2 Inception Module网络结构Figure 2 Inception Module network structure

2.3 多维注意网络结构

将Resnet101的卷积层2和卷积层3融合得到的特征图F3输入一个多维注意力网络，如图3所示。多维注意力网络由像素注意力网络和信道注意力网络组成，用于抑制噪声和突出前景。像素注意力网络将更广泛的上下文信息编码为局部特征，从而增强其表示能力。在这个网络结构中的特征图F3需要进入1个新的Inception Module，这里采用5个3×3的卷积核进行特征的抽取，输出1个两通道的显著特征图。显著特征图分别表示前景和背景的分数。由于显著特征图是连续的，非目标信息不会被完全消除，这有利于保留一定的上下文信息，提高鲁棒性。为了引导该网络学习到目标信息，这里采用了1个有监督学习的方式，使用1个二值化图像作为真实标签，两通道的显著特征图作为预测标签，进而计算2个标签的交叉熵损失，并将其作为注意力损失添加到总损失的计算中。接着在显著特征图上执行softmax操作，并选择一个通道与F3相乘，获得新的信息特征图A3。值得注意的是，在经过softmax函数操作之后的显著特征图的值在[0，1]，说明其可以减少噪声并相对增强对象信息。由于显著特征图是连续的，因此不会完全消除非对象信息，这对于保留某些细节信息并提高鲁棒性是有利的。信道注意力网络通过获取不同通道映射之间的相互依赖性，从而有效增强特征图对于特定语义的表征能力。在这个网络结构中主要采用SE Block[18]的结构，其重点是使用全局平均池化操作，将各个特征图全局感受野的空间信息置入特征图。全局平均池化操作可以通过减少模型中的参数总数来最小化过度拟合。与最大池化层类似，全局平均池化层用于减少三维张量的空间维度。然而全局平均池化层执行更极端的维数减少操作，将尺寸为h×w×d的张量尺寸减小为1×1×d的尺寸。全局平均池化层通过简单地获取h和w的平均值，将每个h×w特征映射层减少为单个数字。每层卷积操作之后都接着一个样例特化激活函数，基于通道之间的依赖关系对每个通道进行一种筛选机制操作，以此对各个通道进行权值评比。

图3 多维注意力网络结构Figure 3 Structure of multidimensional attention network

2.4 损失函数

网络模型在进行训练时需要计算多个位置的损失：RPN阶段的分类和回归损失、Faster R-CNN阶段的分类和回归损失、多维注意力机制阶段的注意力损失，加起来的总损失就是整个网络的损失：

loss=lossrpn+lossfastrcnn+lossattention。

(1)

RPN阶段的分类和回归损失如式(2)所示，其中分类损失lossr_c是判断方框中是否是火灾，也就是区分前景、背景两类物体的损失，如式(3)所示。lossr_l可以对方框位置进行评估和微调，也就是用于比较真实分类的预测参数和真实平移缩放参数的差别，如式(4)所示。lossr_l需要对损失进行L1平滑正则化处理，如式(5)所示，参数σ=1。Faster R-CNN阶段的分类损失计算和RPN阶段一致，而回归损失的L1平滑正则化参数σ=3。

lossrpn=lossr_c+lossr_l。

(2)

(3)

(4)

(5)

(6)

图4 注意力图像Figure 4 Attention image

3 实验结果与分析

本实验选用了开源的Tensorflow框架，该框架可以快速地进行部署和实验，还可以对数据流图进行有效数据可视化，编程语言选用Python3.6，硬件设备CPU为英特尔®酷睿TMi7-7700HQ，内存大小为8 G，GPU型号为GeForce GTX 1060，显存大小为6 G，系统环境为Ubuntu 16.04。

3.1 火灾检测实验数据集

由于目前还没有比较完善的火灾图像检测数据集，本文自主构建了一个火灾图像检测的数据集，总共有210张图片。将所有图片使用标注软件labelImage进行标注，给图像中火灾区域标注1个方框，如图5所示。将标注信息保存到xml文件中，xml文件包括图片文件的位置、名字、宽度width、高度height和维度depth，以及标注方框对应的4个坐标xmin、ymin、xmax和ymax。将图片和标注文件制作成一一对应的tfrecord格式的数据集，然后按7∶3的比例分成训练集和测试集，即训练集中有147张火灾图片，测试集有63张火灾图片，并将2个数据集均转化为tfrecord格式的数据。

图5 标注实例图Figure 5 Example drawing of annotation

3.2 仿真实验

3.2.1 实验过程

实验首先将训练集数据train.tfrecord数据进行解析，得到图片数据和标签，加载卷积网络resnet101.ckpt作为特征提取网络，建立多尺度特征融合的目标检测网络，设置不同的参数来训练数据，使得网络得到最佳的训练效果。对训练集进行8万次迭代得到的损失曲线如图6所示，从图6可以看出，该模型在训练后达到收敛。

图6 损失函数曲线Figure 6 Loss function curve

3.2.2 检测精度评估

在目标检测中，一般使用多种评价指标来进行模型的性能评价。二级指标准确率P为在测试集的识别结果中预测正确的正例数目(TP)占所有预测正例数目(FP+TP)的比例，如式(7)所示。召回率R为在测试集的识别结果中预测正确的正例数目(TP)占所有真实正例数目(TP+FN)的比例，如式(8)所示。利用准确率P和召回率R可以得到P-R曲线，P-R曲线反映了分类器对正例的识别准确程度和对正例的覆盖能力之间的权衡。AP是P-R曲线与X轴围成的图形面积，如式(9)所示。平均检测准确率MAP是对所有类别的AP求均值，如式(10)所示。三级指标F1是准确率和召回率的调和平均值，如式(11)所示。F1指标综合了P与R的产出结果。F1的取值范围为0～1，1代表模型的输出结果最好，0代表模型的输出结果最差。

(7)

(8)

(9)

(10)

(11)

本文所提出的多尺度特征融合网络模型的检测精度指标与其他的模型对比情况如表1所示。其中，所有模型都是以Faster R-CNN模型为基础网络模型，用于对比的模型特征提取网络分别使用了Vgg16、Resnet101、Resnet101+FPN。从表1的对比结果可以发现，使用Resnet101相对于使用Vgg16，MAP值提升了15.2%；使用Resnet101+FPN相对于仅使用Resnet101，MAP值提升了3.6%；使用本文的多尺度特征融合网络模型相对于Resnet101特征提取网络有了8.5%的提升，相对于Resnet101+FPN也有4.9%的提升。多尺度特征融合网络模型的F1相对于Resnet101特征提取网络有2.3%的提高，相对于Resnet101+FPN也有1.7%的提高。

表1 精度评价指标Table 1 Accuracy evaluation index

各个模型对应的P-R曲线如图7所示。从图7可以看出，多尺度特征融合火灾检测算法的P-R曲线下方面积最大，即平均检测准确率MAP最高。

图7 P-R曲线图Figure 7 P-R curve

3.2.3 检测速度评估

由于火灾检测的实时性要求，对于检测速度的评估也是性能评估的一个重要方面。对本文的Resnet101系列模型进行检测速度的评估，最终得到对比结果如表2所示。从表2可以看出，所有的模型均能检测出图像中较为明显的火焰。其中加载模型Faster R-CNN以Resnet101作为特征提取层时，不加上任何改进的检测效率最高，检测1张图片的时间t为0.83 s；使用Resnet101为特征提取层加特征金字塔结构FPN的检测效率最低，检测1张图片的时间t为0.94 s；使用YOLOv3网络[19]的速度最快，时间只需要0.05 s，但是MAP值相对较低为0.762；本文的多尺度特征融合网络得到了速度与精度的平衡，处于中间位置，检测一张图片需0.86 s。

表2 不同方法在火灾检测数据集中的性能对比Table 2 Performance comparison of different methods in fire detection dataset

3.2.4 室内外火灾检测

火灾发生分室内和室外，使用本文多尺度特征融合火灾检测模型，输入图8(a)的室内火灾图像可以得到图8(b)的检测结果，输入图8(c)的室外火灾图像可以得到图8(d)的检测结果。通过对比可以看出，使用本文的多尺度特征融合火灾检测算法能够较好地检测出室内、室外火灾的目标区域。

图8 室内外火灾检测结果Figure 8 Fire image input and output of indoor and outdoor

4 结论

本文首先对火灾检测的现状进行了深入的分析，指出深度学习相对于手工提取特征的优势，并且参考了传统的机器学习的算法思想，在卷积神经网络中加入了多种特征的融合提取。首先将Resnet101的浅层特征和高层特征进行融合，接着使用Inception Module网络提取了多种尺寸的特征，最终加入了多维注意力机制，这些操作使得特征提取网络模型提取特征更加充分，更加接近人的视觉模型。该模型相对原始的Resnet101特征提取网络的检测精度提升了8.5%，相比Resnet101加特征金字塔结构FPN也有4.9%的提升，最终得到平均检测准确率MAP为0.851。对Resnet101特征提取网络进行多种改进操作后检测1张火灾图片的时间为0.86 s。经过验证可知，本文的火灾检测算法能够检测出室内和室外火灾的目标区域，在保证检测速度与原始模型[10]接近的同时，精度相对于原始模型得到较大的提升。