发明公开
- 专利标题: 基于频域注意力机制的烟火图像剪枝蒸馏检测方法
-
申请号: CN202211167078.3申请日: 2022-09-23
-
公开(公告)号: CN115797855A公开(公告)日: 2023-03-14
- 发明人: 杨建旭 , 黄文礼 , 杨可军 , 张可 , 张庚生 , 童旸 , 陆年生 , 茆骥
- 申请人: 南瑞集团有限公司
- 申请人地址: 江苏省南京市江宁经济技术开发区诚信大道19号
- 专利权人: 南瑞集团有限公司
- 当前专利权人: 南瑞集团有限公司
- 当前专利权人地址: 江苏省南京市江宁经济技术开发区诚信大道19号
- 代理机构: 北京清亦华知识产权代理事务所
- 代理商 顾鲜红
- 主分类号: G06V20/52
- IPC分类号: G06V20/52 ; G06V10/42 ; G06V10/44 ; G06V10/764 ; G06V10/82 ; G06N3/0464 ; G06N3/047 ; G06N3/048 ; G06N3/08
摘要:
本申请公开基于频域注意力机制的烟火图像剪枝蒸馏检测方法,涉及电力技术领域,本申请通过增加注意力机制能让模型对图像中输入的变动项更加敏感,频域注意力机制考虑不同的通道特征的频率,获得更多的特征信息,如图像中的烟和火变化剧烈,频域注意力机制则将图像中的频率变化考虑进去,让特征对图像的频率变化敏感,增强其检测精度;同时对模型进行剪枝和知识蒸馏,不仅对特征通道进行裁剪,并基于残差网络,对卷积的kernel长宽进行裁剪,并对裁剪后的网络通过未裁剪的网络模型对其知识蒸馏,使得模型进一步轻量化,提高了模型响应水平,同时保证精度维持在原有水平。