基于视觉关联增强的视觉目标关系抽取方法

本发明公开了基于视觉关联增强的视觉目标关系抽取方法，所述方法包括：构建视觉目标关系抽取模型；计算所有目标对的视觉关联向量；把所有目标的特征表示和所有目标对的视觉关联向量输入特征增强层，获得所有目标的增强后的特征表示；把目标对的增强后的特征表示输入关系预测层，获得预测的关系概率向量；计算并优化关系抽取损失函数，训练视觉目标关系抽取模型，用视觉目标关系抽取模型进行视觉目标关系抽取。本发明通过计算计算主体特征表示和客体特征表示，利用主体特征表示和客体特征表示计算预测的关系概率向量，最后用难例关注损失函数和二范数损失函数的联合损失函数获得高精度的视觉目标关系抽取。