火焰识别方法、系统以及用于识别火焰的神经网络
摘要:
本发明实施方式提供一种火焰识别方法、系统以及用于识别火焰的神经网络,属于图像识别技术领域。所述方法包括:从图像中提取关于火焰的深度特征;根据火焰的颜色模型生成图像掩膜;根据图像掩膜在图像中形成多个火焰可能存在的区域,并在区域上形成多个矩形框;确定矩形框中是否存在物体;在确定存在物体的情况下对矩形框的尺寸进行修正;将修正后的矩形框映射到深度特征中的相应区域,进行下采样处理以得到特征向量;采用全连接层依据特征向量判断每个矩形框中是否存在火焰;在判断矩形框中存在火焰的情况下,修正矩形框;采用神经网络单元判断图像中是否存在火焰;在判断存在火焰的情况下,输出进一步修正后的矩形框作为识别的火焰的位置。
0/0