一种基于注意力机制实现端到端的森林火灾检测算法制造技术

本发明专利技术公开了一种基于注意力机制实现端到端的森林火灾检测算法，包括步骤S1：收集森林火灾事件系列图片，并对其进行预处理；S2：将森林火灾事件系列图片的特征分为浓烟与火焰两个特征表现；根据相对应的特征表现，依次为第一数据集中拼接图赋予第一标签与第二标签进行标注；S3：采用卷积神经网络对拼接图提取2D特征，并采用交叉注意力机制将提取出的2D特征进行特征融合；S4：采用标准COCO数据集进行预训练，并通过迁移学习将其迁移至第一数据集进行训练；按比例划分为训练集与测试集，将训练集送入网络模型中进行迭代训练，得到最优模型；S5：将测试集中的拼接图送入训练好的最优模型中，确认拼接图目标中的类别及位置，输出最终结果。最终结果。最终结果。

【技术实现步骤摘要】
一种基于注意力机制实现端到端的森林火灾检测算法


[0001]本专利技术涉及计算机图像识别
，具体涉及一种基于注意力机制实现端到端的森林火灾检测算法。

技术介绍

[0002]森林火灾不仅会对自然环境造成严重的危害，而且会对人民财产安全造成不可估量的损害。如何通过有效的手段做到及时预警，及时发现，及时处理成为当前火灾检测领域一个至关重要的研究方向。目前火灾检测主要通过以下两种方式：一种是通过红外相机和温度传感器组合的方式进行检测。当火灾发生时，由于局部环境温度会升高，通过红外相机和温度传感器可以有效的检测出环境温度的变化，从而可以判断出是否有火灾发生。另一种方式是通过基于深度学习的目标检测手段，对大量火灾数据进行训练，从而可以达到对火灾的识别和定位。第一种方法很大程度上受制于红外相机和温度传感器的安装环境，比如发电锅炉及炼钢厂附近，那么该设备的准确率将大大降低。第二种方法虽然简单有效，但是却需要提前手工设计Anchor机制，这对于非计算机视觉人员来说是非常困难的；其次目前主流的目标检测算法，还不能实现真正意义上的端到端的处理，一般都需要一个后本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于注意力机制实现端到端的森林火灾检测算法，其特征在于，包括步骤：S1：收集森林火灾事件系列图片，并对其进行预处理；先对系列图片进行数据增强，之后按照随机裁剪与随机排布的方式进行拼接生成拼接图，将多个生成的拼接图进行组合生成第一数据集，并对第一数据集引入预设高斯噪声；S2：将森林火灾事件系列图片的特征分为浓烟与火焰两个特征表现；根据相对应的特征表现，依次为第一数据集中拼接图赋予第一标签与第二标签进行标注；并将标注后的拼接图按照其特征表现的标签，生成与该图片对应的标签组团，并按序显示其标签类别；S3：采用卷积神经网络对拼接图提取2D特征，并采用交叉注意力机制将提取出的2D特征进行特征融合；其预测头采用前向网络FFN对拼接图的类型信息与定位框进行预测，其中定位框采用匈牙利算法进行二分图匹配，并对损失函数进行优化；在多次优化后输出唯一定位框；S4：采用标准COCO数据集进行预训练，并通过迁移学习将其迁移至第一数据集进行训练；将第一数据集按比例划分为训练集与测试集，将训练集送入网络模型中进行迭代训练，并采用自适应梯度下降算法调节其学习率，得到最优模型；S5：根据上述步骤S4的训练结果，将测试集中的拼接图送入训练好的最优模型中，确认拼接图目标中的类别及位置，输出最终结果。2.根据权利要求1所述的一种基于注意力机制实现端到端的森林火灾检测算法，其特征在于，在上述步骤S1中，预处理过程采用Mosaic
‑
4方法进行数据增强，并选取4张系列图片进行拼接组合，输出一张拼接图。3.根据权利要求1所述的一种基于注意力机制实现端到端的森林火灾检测算法，其特征在于，所述第一标签与第二标签分别对应浓烟与火焰，其标注为YOLO格式标注；任一一张拼接图经过标注后都会根据其标签类别依次生成的标签组团中包括：标签类别、标记框中心横坐标与图片宽度的比值、标记框中心纵坐标与图片高度的比值、标记框宽度与图片宽度的比值以及标记框高度与图片高度的比值。4.根据权利要求3所述的一种基于注意力机制实现端到端的森林火灾检测算法，其特征在于，在上述步骤S3中，采用CNN骨干网络ResNet
‑
50模型进行特征提取输出特征图；在对图片进行提取之前，还先将ResNet
‑
50模型加入COCO数据集上进行预训练，然后冻结其权重参数，再迁移到目标数据集上进行二次训练后再对拼接图进行2d特征提取。5.根据权利要求4所述的一种基于注意力机制实现端到端的森林火灾检测算法，其特征在于，任一拼接图还添加有位置编码，位置编码采用绝对位置编码；任一拼接...

【专利技术属性】
技术研发人员：李琪林，严平，蔡君懿，叶润，
申请(专利权)人：国网四川省电力公司营销服务中心，
类型：发明
国别省市：