一种基于卷积神经网络的烟火检测方法技术

本发明专利技术公开了一种基于卷积神经网络的烟火检测方法，目的在于搭建一个轻量级，速度快，成本低的烟火检测平台，实现对烟火的高效检测，实现了一种更加轻量级的网络，可以部署于移动设备上，能够缓解人工监控的压力，降低监控成本。本发明专利技术基于人工智能视频分析和深度学习技术，能够不依赖其他传感设备，实现对视频监控区域内的烟雾与火焰的检测与识别，并及时进行预警。本发明专利技术涉及智慧城市智能安防技术领域。域。域。

【技术实现步骤摘要】
一种基于卷积神经网络的烟火检测方法


[0001]本专利技术涉及智慧城市智能安防
，具体是指一种基于卷积神经网络的烟火检测方法。

技术介绍

[0002]随着社会经济的高速发展，工业、企业园区，住宅日益增多，存在一些消防安全隐患，导致火灾事故屡屡发生。针对住宅、工业园区、森林、加油站等室内外场景，通过深度学习相关的目标检测算法，对监控视频中疑似发生烟雾、火焰的场景进行自动检测，及时发出预警，实现机器智能值守，具有重大的应用价值与社会意义。烟雾与火焰在发生初期如果不能被及时发现并通知管理人员采取措施，较大的烟雾会污染输电线路，火势较大时甚至会烧毁输变电设施，造成大规模停电，如果发生在森林将会引起大规模森林火灾，损害人们的生命及财产安全。由于烟火存在发生时间、地点不固定的特点，低效率的人工监控方式和高成本的服务器端图像识别已无法满足现阶段的需求。而使用硬件对特定区域进行检查，又存在成本设计高，检测区域小，反应时间较慢等问题。因此，如何解决对烟雾和火焰进行特定目标检测并及时作出响应，并且降低成本，提高检测精度减少误报率是当前迫切需要解决的问题之一。

技术实现思路

[0003]针对上述情况，为克服现有技术的缺陷，本专利技术提供了一种基于卷积神经网络的烟火检测方法，目的在于搭建一个轻量级，速度快，成本低的烟火检测平台，实现对烟火的高效检测，实现了一种更加轻量级的网络，可以部署于移动设备上，能够缓解人工监控的压力，降低监控成本。本专利技术基于人工智能视频分析和深度学习技术，能够不依赖其他传感设备，实现对视频监控区域内的烟雾与火焰的检测与识别，并及时进行预警。
[0004]本专利技术采取的技术方案如下：一种基于卷积神经网络的烟火检测方法，包括如下步骤：
[0005]步骤1，利用网络爬虫和开源数据集对不同场景下的烟雾和火灾场景图像进行保存，并且采用人工标注的方式对图像中烟雾和火灾场景进行分块标注，保存格式为VOC格式，将标注的图片划分为训练集和验证集；
[0006]步骤2，将步骤一获得的VOC格式标签和图像通过脚本转化给YOLO格式标签的txt文件，并作为初始训练集和验证集；对项目中的初始的数据集进行Mosaic数据增强，自适应锚框计算，其中Mosaic是将图像进行随机缩放，裁剪，排布的方式进行拼接，增加了对小目标火焰和烟雾的检测效果；
[0007]步骤3，通过比例缩放的方式，将每张图像设置为相同的输入尺寸640
×
640作为初始输入数据；
[0008]步骤4，将进行数据增强后的数据集送入改进的YOLOv5s模型对图像中烟雾与火焰的特征提取进行迭代训练；
[0009]步骤5，网络在初始锚框的基础上输出预测框，进而和真实框groundtruth进行比对，计算两者差距，再反向更新，迭代网络参数，最后将步骤4中训练模型的最好权重文件结果进行保存。
[0010]作为本方案的进一步优化，所述步骤2包括如下子步骤：
[0011]步骤2.1：将获取的VOC格式xml文件通过脚本转化为YOLO格式对应的txt文件，其中脚本完成步骤就是将VOC格式标签中图像的实际宽高，标注框的左上角和右下角坐标，进行归一化处理转化为YOLO格式标签中标注框的中心坐标以及标注框的宽和高；
[0012]归一化的中心坐标计算公式(1)如下：
[0013][0014]其中，x,y为图像的坐标；
[0015]归一化框的高和宽计算公式(2)如下:
[0016][0017]步骤2.2：将对数据集使用自适应锚框算法，以真实的边框位置相对于预设边框的偏移来构建；一个Anchor Box由边框的纵横比和边框的面积来定义，相当于一系列预设边框的生成规则，根据Anchor Box，可以在图像的任意位置，生成一系列的边框，其中边框的计算公式如(3)所示：
[0018][0019]作为本方案的进一步优化，所述步骤4包括如下子步骤：
[0020]步骤4.1：将步骤3中进行过Mosaic数据增强处理的图像输入改进后的YOLOv5模型中BackBone层中；在原有YOLOv5的基础上加入了CBAM注意力模块，其中CBAM公式(4)如下：
[0021][0022]步骤4.2：其中被送入BackBone层中的数据将经过CBAM的通道注意力模块进行特征提取，对输入特征图在空间维度分别进行最大池化和平均池化，然后分别经过一个共享权重的MLP，然后将两者的输出做逐元素的相加，再经过sigmoid激活函数，得到通道注意力权重，将该权重和输入特征图做逐元素的乘法；其中权系数M
c
(F)矩阵公式(5)如下：
[0023][0024]其中，σ为Sigmoid激活函数，W0，W1为MLP的权重，W0∈R
C/r
×
C
，W1∈R
C/r
×
C
，并且r＝16为还原比；
[0025]步骤4.3：其中空间注意力模块注重特征位置的区分和通道注意力机制的补充，在通道维度上分别采用平均池化和最大池化，在最大池化和平局池化之后结合和两个特征映射。对7
×
7卷积核进行降维滤波，将信道数减至1；最后利用Sigmoid激活函数得到空间注意特征图；其中空间注意力的权系数M
s
(F)矩阵公式(6)如下：
[0026][0027]步骤4.4：随后将BackBone层提取特征输入至Head层随后输出特征图，同时图像被划分为小方块并生成锚框，将标注的预测框和特征图进行关联，最后建立损失函数并开启端到端的训练，其中损失函数如公式(7)所示：
[0028][0029]其中ρ2(b,b
gt
)代表了预测框和真实框的中心点的欧式距离，c代表的是能够同时包含预测框和真实框的最小闭包区域的对角线距离。v用来衡量两个矩形框相对比例的一致性，α是权重系数：
[0030][0031]作为本方案的进一步优化，所述训练集和验证集的划分比例为8:2。
[0032]采用上述结构本专利技术取得的有益效果如下：本方案基于卷积神经网络的目标检测方法，将空间注意力机制的思想引入到目标检测的算法中，并提出了一种对于烟雾和火焰特征的新型目标检测的方法，增加了算法的精度的同时保持算法的速度，需要的硬件资源开销较小，结合了数据增强方法加强了对特征的识别。对多种场景的烟雾和火焰数据都进行了分析，选取了最适合用于实际生产场景的复杂烟雾和火焰图像，使得模型能学习到较强的特征，增加了模型的抗干扰能力，与其他模型相比，既能够在复杂的场景中准确检测出烟雾和火焰的存在，又兼顾了实际的运行速度，且在动态场景中检测概率极高，具有广阔的应用前景。
附图说明
[0033]附图用来提供对本专利技术的进一步理解，并且构成说明书的一部分，与本专利技术的实施例一起用于解释本专利技术，并不构成对本专利技术的限制。在附图中：
[0034]图1为各种复杂场景下烟雾与火焰的图片的真实标注；
[0035]图2是数据集中烟雾和本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于卷积神经网络的烟火检测方法，其特征在于，包括如下步骤：步骤1，利用网络爬虫和开源数据集对不同场景下的烟雾和火灾场景图像进行保存，并且采用人工标注的方式对图像中烟雾和火灾场景进行分块标注，保存格式为VOC格式，将标注的图片划分为训练集和验证集；步骤2，将步骤一获得的VOC格式标签和图像通过脚本转化给YOLO格式标签的txt文件，并作为初始训练集和验证集；对项目中的初始的数据集进行Mosaic数据增强，自适应锚框计算，其中Mosaic是将图像进行随机缩放，裁剪，排布的方式进行拼接，增加了对小目标火焰和烟雾的检测效果；步骤3，通过比例缩放的方式，将每张图像设置为相同的输入尺寸640
×
640作为初始输入数据；步骤4，将进行数据增强后的数据集送入改进的YOLOv5s模型对图像中烟雾与火焰的特征提取进行迭代训练；步骤5，网络在初始锚框的基础上输出预测框，进而和真实框groundtruth进行比对，计算两者差距，再反向更新，迭代网络参数，最后将步骤4中训练模型的最好权重文件结果进行保存。2.根据权利要求1所述的一种基于卷积神经网络的烟火检测方法，其特征在于，所述步骤2包括如下子步骤：步骤2.1：将获取的VOC格式xml文件通过脚本转化为YOLO格式对应的txt文件，其中脚本完成步骤就是将VOC格式标签中图像的实际宽高，标注框的左上角和右下角坐标，进行归一化处理转化为YOLO格式标签中标注框的中心坐标以及标注框的宽和高；归一化的中心坐标计算公式(1)如下：其中，x,y为图像的坐标；归一化框的高和宽计算公式(2)如下:步骤2.2：将对数据集使用自适应锚框算法，以真实的边框位置相对于预设边框的偏移来构建；一个Anchor Box由边框的纵横比和边框的面积来定义，相当于一系列预设边框的生成规则，根据Anchor Box，可以在图像的任意位置，生成一系列的边框，其中边框的计算公式如(3)所示：。3.根据权利要求1所述的一种基于卷积神经网络的烟火检测方法，其特征在于，所述步...

【专利技术属性】
技术研发人员：王璇，宋永超，吕骏，徐金东，阎维青，王莹洁，赵金东，王金政，
申请(专利权)人：烟台大学，
类型：发明
国别省市：