基于语义空间约束和注意力机制的场景图生成方法

摘要

本发明公开了一种基于语义空间约束和注意力机制的场景图生成方法，包括以下步骤：S1：采用VG数据集，随机选取若干张图像作为验证集，在剩余图像中按比例随机选取，分别作为训练集和测试集；S2：基于深度学习的方法和VG数据集，构建目标检测网络；S3：建立关系推理网络；S4：根据关系推理网络得到的语义约束信息和注意力网络，得到突出可能存在相关关系的区域和通道特征，进行关系推理分类。本发明无需使用图相关信息流传递机制，基于语义空间的约束和注意力机制推测两两物体之间的相关关系，没有引入更多的噪声，网络比较简单，提供了更高的关系推理精度，能够更有效提取出存在物体相关关系的特征。