一种非局域的图像分类装置、方法和存储介质

摘要

本发明公开了一种非局域的图像分类装置、方法和存储介质，所述卷积网络由从前至后依次连接的根部模块、若干个残差模块以及非局域模块、头部模块构成；所述非局域模块由从前至后依次连接的坐标拼接模块、键值生成模块、非局域注意力模块、注意力融合模块构成；所述坐标拼接模块用于将特征图上各特征点的绝对坐标信息加入到特征向量中；所述键值生成模块输出查询向量、键向量和值向量，并输入非局域注意力模块，处理得到注意力输出张量；所述注意力融合模块的输入为非局域模块的输入以及非局域注意力模块的输出。本发明通过非局域模块使输出的特征图上的所有特征点都可以获取到全局域的信息，精度改善明显，有效提升了网络的性能。