烟火检测模型的训练方法、烟火检测方法及设备
基本信息
| 申请号 | CN202110215838.2 | 申请日 | - |
| 公开（公告）号 | CN112949453A | 公开（公告）日 | 2021-06-11 |
| 申请公布号 | CN112949453A | 申请公布日 | 2021-06-11 |
| 分类号 | G06K9/00;G06K9/62;G06N3/04;G06N3/08 | 分类 | 计算；推算；计数; |
| 发明人 | 曹毅超;孙飞;施燕平;李溯;陈斌锋;封晓强 | 申请（专利权）人 | 南京恩博科技有限公司 |
| 代理机构 | 江苏瑞途律师事务所 | 代理人 | 蒋海军 |
| 地址 | 210000 江苏省南京市玄武区长江后街6号东大科技园创业园长江楼B208室 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明公开了一种烟火检测模型的训练方法、烟火检测方法及设备，属于烟火检测技术领域。训练方法包括：构建视频烟火样本数据集；将输入图像输入纹理支路进行特征提取，得到多尺度特征表示，并通过特征金字塔将多尺度特征表示融合为纹理特征；计算输入图像与参考图像的帧差图像，将帧差图像输入运动支路计算运动注意力权重图；对纹理特征进行运动感知增强；利用弱注意力模块生成弱烟火目标掩码；根据运动感知增强后的纹理特征和弱烟火目标掩码，得到烟火特征图，并进行烟火目标检测。并且提出了利用所训练出的模型进行烟火检测的方法，以及执行所述烟火检测方法的设备。本发明能够有效地提高烟火的检测准确率，并且计算成本低，便于部署。 |
