基于DeOldify和YOLOv8的灰度图火焰识别方法、系统及存储介质
申请号:CN202410871693
申请日期:2024-07-01
公开号:CN118865110A
公开日期:2024-10-29
类型:发明专利
摘要
本申请涉及图像识别的技术领域,公开基于DeOldify和YOLOv8的灰度图火焰识别方法、系统及存储介质,方法包括:通过卷积计算提取出灰度图中的火焰特征;对火焰特征解码得到火焰灰度图像,基于自注意力层捕捉特征图像的全局结构信息,将全局结构信息嵌入火焰灰度图像;基于子像素卷积算法对火焰灰度图像进行上采样得到特征图像;将特征图像缩放到归一化RGB空间得到着色图像;使用VGG模型提取灰度图中的火焰模型;基于骨干网络从着色图像中提取出着色特征;基于颈部网络对着色特征进行融合得到火焰融合图像;基于检测头网络从物体融合图像中提取得到火焰的边界框信息和类别信息;本申请能在火焰图片的颜色缺失和夜间或低光照条件下的干扰中准确地识别出火焰。
技术关键词
火焰识别方法
全局结构信息
图像
Sigmoid函数
着色
上采样
卷积算法
网络
火焰识别系统
注意力
编码器
检测头
分支
像素
批量
缩放参数
输出特征
尺寸
解码器