基于频域注意力机制的烟火图像剪枝蒸馏检测方法
摘要:
本申请公开基于频域注意力机制的烟火图像剪枝蒸馏检测方法,涉及电力技术领域,本申请通过增加注意力机制能让模型对图像中输入的变动项更加敏感,频域注意力机制考虑不同的通道特征的频率,获得更多的特征信息,如图像中的烟和火变化剧烈,频域注意力机制则将图像中的频率变化考虑进去,让特征对图像的频率变化敏感,增强其检测精度;同时对模型进行剪枝和知识蒸馏,不仅对特征通道进行裁剪,并基于残差网络,对卷积的kernel长宽进行裁剪,并对裁剪后的网络通过未裁剪的网络模型对其知识蒸馏,使得模型进一步轻量化,提高了模型响应水平,同时保证精度维持在原有水平。
0/0