基于空间关联增强的目标关系抽取方法

本发明公开了基于空间关联增强的目标关系抽取方法，所述方法包括：构建视觉关系抽取模型，包括预训练视觉模型、特征增强层和关系预测层；使用预训练视觉模型提取图片特征；根据图片特征和目标区域标注获得所有目标的特征表示；计算所有目标对的空间关联向量；把所有目标的特征表示和所有目标对的空间关联向量输入特征增强层，获得所有目标的增强后的特征表示；把目标对的增强后的特征表示输入关系预测层，获得预测的关系概率向量；计算并优化关系抽取损失函数，训练视觉关系抽取模型，用视觉关系抽取模型进行视觉关系抽取。本发明对空间关联进行显式建模，创新性地用空间关联引导目标特征进行聚合，增强目标的特征，实现高精度的视觉关系抽取。