基于几何属性引入的目标关系分类方法

本发明公开了基于几何属性引入的目标关系分类方法，所述方法包括：构建视觉关系分类模型，包括预训练视觉模型和关系预测层；使用预训练视觉模型提取图片特征；根据图片特征和目标区域标注获得所有目标的特征表示；计算目标对的几何属性向量；把目标对的特征表示和目标对的几何属性向量输入关系预测层，获得预测的关系概率向量；计算并优化关系分类损失函数，训练视觉关系分类模型，用视觉关系分类模型进行视觉关系分类。本发明对几何属性，包括面积属性、方向属性，进行显式建模，并将这种几何属性显式地引入视觉关系分类模型，使模型能够感知到目标间的先验几何属性，实现高精度的视觉关系分类。