视觉关系检测方法和装置及视觉关系检测训练方法和装置

摘要

本申请提供了视觉关系检测方法及装置、以及视觉关系检测训练方法及装置。视觉关系检测方法包括：从图像提取用于物体检测的多个候选区域；基于多个候选区域获得多个视觉关系检测区域集，视觉关系检测区域集包括主语检测区域、宾语检测区域及谓语检测区域；以及分别基于每个视觉关系检测区域集中的主语检测区域、宾语检测区域及谓语检测区域，确定主语物体、宾语物体及谓语关系。通过本申请提供的方法和装置，能够将视觉关系中的主语、谓语和宾语三个部分作为整体来考虑，同时识别出这三个部分，提高了视觉关系检测的准确率。