一种基于可见光和红外特征融合网络的火焰检测方法

    公开(公告)号:CN118799718A

    公开(公告)日:2024-10-18

    申请号:CN202310388481.7

    申请日:2023-04-12

    Abstract: 本发明公开了一种基于可见光和红外特征融合网络的火焰检测方法,包括:建立火焰数据集、以DETR模型为基础模型,改造提取特征的主干网络为对称结构、在特征提取网络和编码解码结构之间添加特征融合网络得到第一Fire‑DETR模型、将处理好的训练集输入第一Fire‑DETR模型进行训练优化得到第二Fire‑DETR模型、将测试图像输入第二Fire‑DETR模型得到预测结果。称的特征提取网络使检测模型可以同时提取到可见光和红外两种特征信息用以预测,扩展了火焰特征图的多样性。添加的特征融合网络能够考虑通道间关系和空间信息,使得提取到的不同特征可以有选择性地进行融合。与现有的目标检测深度卷积网络相比,改进后的Fire‑DETR模型对小面积火焰和噪音的鲁棒性更强,且火焰的检测精度更高。

Patent Agency Ranking