基于空间关系判别的目标关系识别方法

本发明公开了基于空间关系判别的目标关系识别方法，所述方法包括：构建视觉关系抽取模型，包括预训练视觉模型、空间关系判别模块和语义关系预测层；使用预训练视觉模型提取图片特征；根据图片特征和目标区域标注获得所有目标的特征表示；把目标对的特征表示输入空间关系判别模块，获得预测的空间关系概率向量，计算空间关系判别损失函数；把目标对的特征表示输入语义关系预测层，获得预测的关系概率向量，计算关系抽取损失函数；计算并优化联合损失函数，训练视觉关系抽取模型，用该模型进行视觉关系抽取。本发明使用多任务学习的思路，创新性地提出空间关系判别的辅助任务，让视觉关系抽取模型学习到空间关系，实现高精度的视觉关系抽取。