一种提取整合图像内部与图像之间特征关系的方法和装置

摘要

本发明公开了一种提取整合图像内部与图像之间特征关系的方法及装置。该方法包括如下步骤：使用内部关系网络表征图像内部的特征关系，使用交叉关系网络表征图像之间的特征关系；其中，将内部关系网络划分为空间关系模块和通道关系模块；将图像序列分别输入空间关系模块、通道关系模块和交叉关系网络；在空间关系模块、通道关系模块和交叉关系网络中，图像序列依次经过压缩层、基于注意力的关系模块层和恢复层的处理之后，进入批量规范化层以加速卷积神经网络的收敛速度；使用卷积融合函数，将空间关系模块、通道关系模块和交叉关系网络的输出融合在一起，作为输出结果。利用本发明，可以增强卷积神经网络在抽象视觉推理任务中的推理能力。