基于图结构的图像注意力网络

Graph attention network based on graphic structure

虽然现已有许多关于图像注意力机制的研究,但是现有的方法往往忽视了特征图的全局空间结构和空间注意力与通道注意力的联系。所以本文提出了一种基于整个空间拓扑结构的注意机制,将特征图映射成结点与特征的形式,再借助图卷积网络的特性,得以从整个空间学习特征权重图。其次空间注意力与通道注意力一体化结构能够更有效地学习特征权重。通过多个实验测试表明,在图像分类和人脸识别任务中,展现了优异的性能和普遍适用性。