森林火灾源头的估计模型训练方法、估计方法及系统

基本信息

申请号 CN202110097330.7 申请日 -
公开(公告)号 CN112836608A 公开(公告)日 2021-05-25
申请公布号 CN112836608A 申请公布日 2021-05-25
分类号 G06K9/00;G06K9/62;G06N3/04;G06N3/08 分类 计算;推算;计数;
发明人 曹毅超;刘建水;李彦;陈斌锋;李溯;封晓强 申请(专利权)人 南京恩博科技有限公司
代理机构 江苏瑞途律师事务所 代理人 蒋海军
地址 210000 江苏省南京市玄武区长江后街6号东大科技园创业园长江楼B208室
法律状态 -

摘要

摘要 本发明公开了一种森林火灾源头的估计模型训练方法、估计方法及系统,属于计算机视觉和森林烟火视频监测领域。本发明的估计模型包括主干网络、前景特征增强模块、多帧特征融合模块,训练模型时包括构建森林火灾视频样本数据集;对正样本和负样本进行预处理;将经过预处理后的森林火灾视频样本输入主干网络提取图像的高维特征,将高维特征输入前景特征增强模块增强前景特征,输入多帧特征融合模块进行特征融合,融合后进行反卷积操作,将高维特征映射为火灾源头的估计掩码;利用Focal Loss损失函数进行模型参数优化。估计方法采用所训练的模型对森林火灾源头进行估计,并将估计方法部署于估计系统中。本发明能够准确地对森林火灾源头进行识别和估计。