一种基于神经网络频域注意力机制的图像识别方法

摘要

本发明公开了一种基于神经网络的频域注意力机制设计方法，用于图像识别。具体包括如下步骤：获取用于训练神经网络的图像识别数据集，定义算法目标；建立单一频域变换基函数选择模型；建立组合频域变换基函数选择模型；建立基于神经网络的频域注意力机制；基于前述的建模结果训练预测模型；使用所述预测模型的进行图像识别。本发明通过将不同频域的信息纳入注意力机制中，实现了在多种图像识别任务(图像分类、目标检测、实例分割)上、相同计算量与复杂度情况下精度的大幅度提升，具有良好的应用价值。