本发明专利技术公开了一种基于u2net的烟雾、火焰分割方法及系统，分割方法的步骤包括：构建具有三通道的烟雾、火焰分割网络，所述烟雾、火焰分割网络包括三通道的u2net输出模块、三类的u2net误差模块；构建烟雾、火焰分割模型训练据集；获取烟雾、火焰分割模型；所述待检测图像中烟雾、火焰的概率识别；所述待检测图像中的烟雾、火焰的精准识别等步骤。利用基于u2net的分割模型，得到一个三通道的输出结果，每个通道表示此像素的属于别属于背景、烟雾和火焰的概率；采用log+softmax函数对三通道的输出结果在通道轴做处理，实现图像中背景、烟雾、火焰的多重识别；本发明专利技术方法满足开阔大型空间场所的需求，在大范围的视野中依然能够稳定的鲁棒性，识别精度高。识别精度高。识别精度高。
【技术实现步骤摘要】
一种基于u2net的烟雾、火焰分割方法及系统
[0001]本专利技术涉及于图像监控与处理
，尤其涉及一种基于u2net的烟雾、火焰分割方法及系统。
技术介绍
[0002]火灾是当今社会人类面临的主要灾害之一，火灾的发生严重影响社会经济发展，破坏自然生态，也会给人类的生命和财产带来损失，因此早期火灾检测的意义重大。
[0003]早期的火灾检测主要基于各种传感器的探测，对被探测区域的透明度、光谱、温度和烟粒浓度等进行采样，以确定是否发生火灾。然而，由火灾引起的烟雾浓度、温度和其他信息的变化在空气中传播缓慢，且稳定性不高，必须安装在靠近火源的地方，且受气流环境、探测距离、热障效应等因素影响，很难及时快速地捕获火灾信号，无法在恶劣的室外场所使用，并且无法满足开阔大型空间场所的需求。
[0004]随着科学技术的不断突破，火灾检测技术正在逐步向图像化、智能化发展。对火灾进行实时监控，使火灾造成的损失降到最低是火灾防治
的重点研究内容。
[0005]Toreyin提取火焰的边界轮廓，利用小波变换对边界进行检测，利用火焰的高频特性分析火焰的闪烁特征。Yamagishi和Yamaguch等提出了一种利用彩色摄像机检测视频火焰的算法，提取视频中明显的火焰区域，形成含有火焰的图像块，利用神经网络对图像块的HSV空间颜色进行训练，训练结果对输入图像进行处理后，可以得到图像中是否有火灾，该算法只考虑火焰的颜色特性，不考虑火焰的运动特性，导致误判率高，算法复杂，不能满足实时性。贾阳、喻润洋提出使用U
‑
Net结构的深度神经网络进行早期火灾烟雾的自动分割，通过半自动算法人工辅助分割出烟雾区域的图像样本，基于深度神经网络对分割烟雾区域进行学习，得到原始视频帧到分割结果的映射模型，并据此模型进行烟雾区域分割，该模型基于人工辅助分割实现烟雾区域识别。
[0006]一种火焰和烟雾异常情况检测方法及系统(专利号：201911358605.7)公开了基于颜色特征、小波分析和卷积神经网络相结合的火灾检测方法，采用颜色分割的方法来识别火焰区域，利用背景分块差分模型提取烟雾图像运动像素，对烟雾遮挡物体的特性进行判别，从而检测出烟雾区域，利用卷积神经网络实现火焰、烟雾的具体位置。本方案算法复杂，同时基于颜色分割的火焰识别方案，在大范围的视野中识别精度不高，识别方案鲁棒性不高。
[0007]有鉴于此，特提出本申请。
技术实现思路
[0008]本专利技术的专利技术目的是针对现有技术的不足，提供了一种基于u2net的烟雾、火焰分割方法及系统，实现高精度的火焰及烟雾识别。
[0009]本专利技术为实现上述专利技术目的采用如下技术方案：
[0010]本申请提供一种基于u2net的烟雾、火焰分割方法，包括以下步骤：
[0011]S1：构建具有三通道的烟雾、火焰分割网络，所述烟雾、火焰分割网络包括三通道的u2net输出模块、三类的u2net误差模块；
[0012]S2：构建烟雾、火焰分割模型训练据集，所述烟雾、火焰分割模型训练据集中每一张样本图像的标签为单通道的图像，所述标签中的某坐标位置的像素值表示对应的样本图像中对应坐标位置的像素值的类别，所述类别为背景、烟雾和火焰中的一类；
[0013]S3：获取烟雾、火焰分割模型，采用所述烟雾、火焰分割模型训练据集对所述烟雾、火焰分割网络进行训练，获得所述烟雾、火焰分割模型，所述烟雾、火焰分割模型具有三个输出通道，三个输出通道的输出结果分别为图像中每个坐标的像素值属于背景的概率、属于烟雾的概率、属于火焰的概率；
[0014]S4：采集待检测图像；
[0015]S5：所述待检测图像中烟雾、火焰的概率识别，将所述待检测图像输入所述烟雾、火焰分割模型得到所述待检测图像的所述三个输出通道的输出结果；
[0016]S6：所述待检测图像中的烟雾、火焰的精准识别，对所述待检测图像的所述三个输出通道的输出结果在通道轴做处理，得到三通道的目标特征图，根据所述目标特征图确定所述待检测图像每个坐标的识别结果。
[0017]进一步地，所述三通道的u2net输出模块的构建方法为，将u2net网络的side模块的输出设置为3。
[0018]进一步地，所述三类的u2net误差模块的损失函数为多类交叉熵损失和多类样本相似度度量损失。
[0019]进一步地，所述多类交叉熵损失为：
[0020][0021]其中，Log是以e为底数，i为0，1，2，out[i]为所述烟雾、火焰分割网络的输出向量。
[0022]进一步地，所述多类样本相似度度量损失为：
[0023][0024]其中，TP：true positive，真阳性；TN：true negative，真阴性；FP：false positive，假阳性；FN：false negative，假阴性。
[0025]进一步地，标签的制作过程为：
[0026]定义一张和样本图像大小一样的单通道图像，若所述样本图像中坐标位置的像素值为背景，所述单通道图像中对应的坐标位置处的像素值为0；若所述样本图像中坐标位置的像素值为烟雾，所述单通道图像中对应的坐标位置处的像素值为1；若所述样本图像中坐标位置的像素值为火焰，所述单通道图像中对应的坐标位置处的像素值为2，在所述单通道图像中完成所述样本图像每个坐标位置的像素值的标注，获得所述标签。
[0027]进一步地，在所述S3中，采用梯度下降的方式训练所述烟雾、火焰分割网络。
[0028]进一步地，在所述S6中，采用log+softmax函数对所述待检测图像的所述三个输出通道的输出结果在通道轴做处理。
[0029]所述log+softmax函数为：
[0030][0031]式中：i为0，1，2，p0、p1、p2，分别为图像中某坐标的像素值属于背景的概率、属于烟雾的概率、属于火焰的概率。
[0032]进一步地，所述根据所述目标特征图确定所述待检测图像每个坐标的识别结果的具体步骤为：
[0033]提取所述目标特征图每个坐标的三通道结果，取每个坐标的所述三通道结果的最大值得到每个坐标的最大值，根据每个坐标的所述最大值所在的通道对应的类别获得所述检测图像每个坐标的像素值的类别，所述类别为背景、烟雾和火焰中的一类。
[0034]第二方面，本专利技术提供一种基于u2net的烟雾、火焰分割系统，包括：
[0035]烟雾、火焰分割模型训练据集标注模块；所述烟雾、火焰分割模型训练据集中每一张样本图像的标签为单通道的图像，所述标签中的某坐标位置的像素值表示对应的样本图像中对应坐标位置的像素值的类别，所述类别为背景、烟雾和火焰中的一类；
[0036]烟雾、火焰分割模型训练模块；采用所述烟雾、火焰分割模型训练据集对烟雾、火焰分割网络进行训练，获得所述烟雾、火焰分割模型；
[0037]所述烟雾、火焰分割网络包括三通道的u2net输出模块、三类的u2net误差模块；
[0038]所述本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种基于u2net的烟雾、火焰分割方法，其特征在于，包括以下步骤：S1：构建具有三通道的烟雾、火焰分割网络，所述烟雾、火焰分割网络包括三通道的u2net输出模块、三类的u2net误差模块；S2：构建烟雾、火焰分割模型训练据集，所述烟雾、火焰分割模型训练据集中每一张样本图像的标签为单通道的图像，所述标签中的某坐标位置的像素值表示对应的样本图像中对应坐标位置的像素值的类别，所述类别为背景、烟雾和火焰中的一类；S3：获取烟雾、火焰分割模型，采用所述烟雾、火焰分割模型训练据集对所述烟雾、火焰分割网络进行训练，获得所述烟雾、火焰分割模型，所述烟雾、火焰分割模型具有三个输出通道，三个输出通道的输出结果分别为图像中每个坐标的像素值属于背景的概率、属于烟雾的概率、属于火焰的概率；S4：采集待检测图像；S5：所述待检测图像中烟雾、火焰的概率识别，将所述待检测图像输入所述烟雾、火焰分割模型得到所述待检测图像的所述三个输出通道的输出结果；S6：所述待检测图像中的烟雾、火焰的精准识别，对所述待检测图像的所述三个输出通道的输出结果在通道轴做处理，得到三通道的目标特征图，根据所述目标特征图确定所述待检测图像每个坐标的识别结果。2.根据权利要求1所述的一种基于u2net的烟雾、火焰分割方法，其特征在于，所述三通道的u2net输出模块的构建方法为，将u2net网络的side模块的输出设置为3。3.根据权利要求2所述的一种基于u2net的烟雾、火焰分割方法，其特征在于，所述三类的u2net误差模块的损失函数为多类交叉熵损失和多类样本相似度度量损失。4.根据权利要求3所述的一种基于u2net的烟雾、火焰分割方法，其特征在于，所述多类交叉熵损失为：其中，Log是以e为底数，i为0，1，2，out[i]为所述烟雾、火焰分割网络的输出向量。5.根据权利要求3所述的一种基于u2net的烟雾、火焰分割方法，其特征在于，所述多类样本相似度度量损失为：其中，TP：true positive，真阳性；TN：true negative，真阴性；FP：false positive，假阳性；FN：false negative，假阴性。6.根据权利要求1所述的一种基于u2net的烟雾、火焰分割方法，其特征在于，标签的制作过程为：定义一张和样本图像大小一样的单通道图像，若所述样本图像中坐标位置的像素值为背景，所述单通道图像中对应的坐标位置处的像素值为0；若所述样本图像中坐标...
【专利技术属性】
技术研发人员：李继凯，
申请(专利权)人：北京优创新港科技股份有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。