有效

基于注意力机制的特征加强方法、装置及图像处理方法

谢毓湘、宫铨志、闫洁、魏迎梅、栾悉道、蒋杰、康来、邹诗苇、李竑赋
中国人民解放军国防科技大学

摘要

本发明公开了一种基于注意力机制的特征加强方法、装置及图像处理方法,特征加强方法包括:获取图像待加强的初始特征矩阵;对初始特征矩阵进行特征强化处理,得到特征强化注意力加权后的第一特征矩阵;对初始特征矩阵进行特征空间相关性的提取和处理,得到空间相关性注意力加权后的第二特征矩阵;对初始特征矩阵进行特征通道相关性的提取和处理,得到通道相关性注意力加权后的第三特征矩阵;将第一特征矩阵、第二特征矩阵、第三特征矩阵加权,得到加强特征矩阵。本发明可应用于轻量级卷积神经网络性能的提高,在保证轻量级卷积神经网络参数轻量级的前提下提高图像分类、目标检测、行为识别及实例分割等任务的准确性,满足应用中的基本需求。