基于YCrCb和LAB颜色空间的火焰检测识别方法及系统技术方案

本发明专利技术公开了一种基于YCrCb和LAB颜色空间的火焰检测识别方法及系统，适用于对视频中的火焰实时检测识别。该方法通过公开的火焰视频数据，对CNN网络进行训练，得到用于检测识别视频图像中火焰区域的模型；然后将摄像机获取的视频帧图像从RGB转化为YCrCb颜色空间和LAB颜色空间，在通过灰度化以及高斯滤波的后进行做差，对差值图像二值化后进行开闭运算并提取轮廓图得到火焰候选目标；再利用CNN模型对火焰候选目标进一步识别，按照识别置信度进行删除操作得到火焰目标；最后记录并标记火焰目标在原视频帧中的位置。本发明专利技术可用于实时检测并识别视频中的及其位置，识别率达到93.51％，火焰检测识别速度保持在27‑35帧/s。

【技术实现步骤摘要】
基于YCrCb和LAB颜色空间的火焰检测识别方法及系统
本专利技术涉及图像处理及预防火灾
，具体涉及一种基于YCrCb和LAB颜色空间的火焰检测识别方法及系统。
技术介绍
火焰检测识别分很多种，基于传感器的传统火灾检测方法，检测范围有限且信息单一，检测速度存在较大的延迟。随着计算机和图像处理领域的理论及技术不断突破，研究人员利用机器学习算法结合图像处理来实现火焰的检测。此类方法主要利用火焰的色彩属性等静态特征和火焰圆形度、尖角、轮廓变化、闪动特征等动态特征来进行识别，然而根据先验知识设计人工特征提取算法比较困难和耗时，在面对不同的复杂环境和多变的火焰类型时，其泛化能力往往不足。还有基于超像素火焰检测识别的方法，但存在无法进行实时检测识别的问题。
技术实现思路
专利技术目的：针对传统火焰检测识别存在易受干扰，误检率高的问题，以及在深度神经网络下实时性差的问题，本专利技术目的在于提供一种基于YCrCb和LAB颜色空间的火焰检测识别方法及系统，通过颜色空间结合CNN神经网络的方法，提高火焰检测识别的精确度并且解决实时性检测问题。技术方案：为实现上述专利技术目的，本专利技术采用如下技术方案：一种基于YCrCb和LAB颜色空间的火焰检测识别方法，包括以下步骤：(1)获取视频帧图像，将视频帧图像分别转换为YCrCb和LAB颜色空间图像并灰度化后得到两幅图像；(2)将步骤(1)得到的两幅图像相减得到差值图像，并进行二值化；(3)对二值化后的图像通过开闭运算减少图像的噪点，再通过寻找二值化图像的轮廓图得到火焰候选区域的具体位置；(4)找到火焰候选区域在原视频帧中对应位置的火焰候选区域，使用训练好的CNN模型对该区域目标图像进一步识别，剔除非火焰目标得到火焰目标，并在原视频帧中对火焰目标进行标记。进一步地，训练CNN模型的样本数据集通过对现有公开的火焰视频数据进行处理得到，具体为：通过对公开的火焰视频使用取帧法将视频提取出视频帧集合，并对视频帧集合提取火焰区域与非火焰区域，并将图像分别放入火焰图像和非火焰图像两个不同文件夹；将火焰图像文件夹标签设为正监督值，非火焰图像文件夹标签设为负监督值，再将图像归一化成同一大小构建成CNN模型的样本数据集。进一步地，所述步骤(1)中将YCrCb和LAB颜色空间图像灰度化后再通过高斯滤波减少图像的噪点。进一步地，所述步骤(2)中二值化的阈值范围为8-12。进一步地，所述步骤(3)包括以下步骤：(31)对二值化后的图像进行开运算，去除独立的噪点；(32)对开运算后的图像进行闭运算，弥补相邻目标之间的缝隙；(33)通过寻找轮廓方法得到目标的候选区域。进一步地，所述步骤(4)包括以下步骤：(41)找到二值图像的火焰候选目标在原视频帧中对应的火焰候选目标；(42)将所有的火焰候选目标归一化为统一的大小；(43)送入训练好的CNN模型中进行识别，根据置信度大小剔除非火焰目标得到火焰目标；(44)将经过CNN模型识别为火焰目标的区域对应在原图像的位置通过矩形在视频帧中进行标记，实现火焰追踪的可视化。进一步地，当视频中的一帧图像或连续设定帧数的图像中识别出火焰目标则发出火焰警报。基于相同的专利技术构思，本专利技术所述的一种基于YCrCb和LAB颜色空间的火焰检测识别系统，包括：图像预处理模块：用于读取视频帧图像，将视频帧图像分别转换为YCrCb和LAB颜色空间图像并灰度化后得到两幅图像；火焰检测模块：用于将图像预处理模块得到的两幅图像做差得到差值图像并二值化后，再经过开闭运算、寻找轮廓方法得到火焰目标的候选区域；火焰识别模块：用于使用训练好的CNN模型对得到的火焰目标候选区域图像进行进一步识别，剔除非火焰目标得到火焰目标；以及，火焰区域可视化模块：用于根据火焰识别模块得到的火焰目标位置，对原视频图像中对应的火焰目标进行标记，实现火焰追踪的可视化。进一步地，所述系统还包括火焰警报模块：用于对视频持续监测，在一帧图像或连续设定帧数的图像中识别出火焰目标时，发出警报。基于相同的专利技术构思，本专利技术所述的一种基于YCrCb和LAB颜色空间的火焰检测识别系统，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述计算机程序被加载至处理器时实现所述的基于YCrCb和LAB颜色空间的火焰检测识别方法。有益效果：与现有技术相比，本专利技术的有益效果如下：1、采用CNN模型架构，有效克服了特征提取难，识别速度低的问题。2、采用YCrCb与LAB颜色空间结合的方式识别火焰，更加易于提取火焰位置，提高了对火焰识别的速度以及准确率。3、本专利技术的火焰检测识别速度可保持在27-35帧/s，同时识别率达到93.51％。4、本专利技术可用于实时检查，能够搭载载具进行火焰检测，应用场景更广。附图说明图1为本专利技术实施例的方法流程图；图2为本专利技术实施例中示例图像YCrCb颜色空间的灰度图；图3为本专利技术实施例中示例图像LAB颜色空间的灰度图；图4为本专利技术实施例中火焰候选区域的二值图；图5为本专利技术实施例中火焰检测识别定位的可视化效果图。具体实施方式下面结合附图对本专利技术作进一步的说明。现将本实施例中涉及的各个变量作如下说明，如表1所示。表1变量说明表本专利技术实施例使用的视频数据是MIVIA实验室公开的和网络收集的火焰视频库数据，视频包含不同颜色的火焰，不同形状的火焰，小目标火焰，特殊火焰，与火焰颜色接近的灯光，本方法对该视频数据进行处理，检测识别火焰位置并进行可视化。下面首先说明一下本专利技术实施例中涉及的CNN模型训练过程。将MIVIA实验室公开的和网路收集的火焰视频进行预处理后，训练CNN模型M1。具体包括：1)通过对MIVIA实验室公开的和网络收集的火焰视频使用取帧法将视频每三十帧截取一帧，截取火焰部分与非火焰部分构建训练数据集P1，P1为由火焰图像与非火焰图像组成，其中火焰图像放在Fire文件夹中，非火焰图像放在NoFire文件夹中，每张图像都归一化为64×64像素大小。2)将P1作为CNN的训练集，通过PyTorch的ImageFolder()函数读取存放文件夹Fire和NoFire，并将火焰图像监督值设为正监督值，非火焰图像的监督值设为负监督值，得到标注后的数据集DataSet。3)批训练的EPOCH设为30，BACH_SIZE设为50。4)预训练的权重设为随机值，将CNN网络的输入维度设为(64，64，3)。5)设置CNN网络参数，包括：使用Adam梯度下降法并将学习率设为1×10-4、损失函数设为交叉熵函数。7)将DataSet在CNN网络中训练得到模型M1。如图1所示，本专利技术实施例公开的一种基于YCrCb和LAB颜色空间的火焰检测识别方法，包括如下步骤：(1)从摄像机采集的本文档来自技高网...

【技术保护点】
1.一种基于YCrCb和LAB颜色空间的火焰检测识别方法，其特征在于，包括以下步骤：/n(1)获取视频帧图像，将视频帧图像分别转换为YCrCb和LAB颜色空间图像并灰度化后得到两幅图像；/n(2)将步骤(1)得到的两幅图像相减得到差值图像，并进行二值化；/n(3)对二值化后的图像通过开闭运算减少图像的噪点，再通过寻找二值化图像的轮廓图得到火焰候选区域的具体位置；/n(4)找到火焰候选区域在原视频帧中对应位置的火焰候选区域，使用训练好的CNN模型对该区域目标图像进一步识别，剔除非火焰目标得到火焰目标，并在原视频帧中对火焰目标进行标记。/n

【技术特征摘要】
1.一种基于YCrCb和LAB颜色空间的火焰检测识别方法，其特征在于，包括以下步骤：
(1)获取视频帧图像，将视频帧图像分别转换为YCrCb和LAB颜色空间图像并灰度化后得到两幅图像；
(2)将步骤(1)得到的两幅图像相减得到差值图像，并进行二值化；
(3)对二值化后的图像通过开闭运算减少图像的噪点，再通过寻找二值化图像的轮廓图得到火焰候选区域的具体位置；
(4)找到火焰候选区域在原视频帧中对应位置的火焰候选区域，使用训练好的CNN模型对该区域目标图像进一步识别，剔除非火焰目标得到火焰目标，并在原视频帧中对火焰目标进行标记。


2.根据权利要求1所述的一种基于YCrCb和LAB颜色空间的火焰检测识别方法，其特征在于，训练CNN模型的样本数据集通过对现有公开的火焰视频数据进行处理得到，具体为：通过对公开的火焰视频使用取帧法将视频提取出视频帧集合，并对视频帧集合提取火焰区域与非火焰区域，并将图像分别放入火焰图像和非火焰图像两个不同文件夹；将火焰图像文件夹标签设为正监督值，非火焰图像文件夹标签设为负监督值，再将图像归一化成同一大小构建成CNN模型的样本数据集。


3.根据权利要求1所述的一种基于YCrCb和LAB颜色空间的火焰检测识别方法，其特征在于，所述步骤(1)中将YCrCb和LAB颜色空间图像灰度化后再通过高斯滤波减少图像的噪点。


4.根据权利要求1所述的一种基于YCrCb和LAB颜色空间的火焰检测识别方法，其特征在于，所述步骤(2)中二值化的阈值范围为8-12。


5.根据权利要求1所述的一种基于YCrCb和LAB颜色空间的火焰检测识别方法，其特征在于，所述步骤(3)包括以下步骤：
(31)对二值化后的图像进行开运算，去除独立的噪点；
(32)对开运算后的图像进行闭运算，弥补相邻目标之间的缝隙；
(33)通过寻找轮廓方法得到目标的候选区域。


6.根据权利要...

【专利技术属性】
技术研发人员：高尚兵，王雨豪，相林，朱全银，张海艳，李翔，丁海林，袁涛，
申请(专利权)人：淮阴工学院，
类型：发明
国别省市：江苏;32