[发明专利]场景图生成方法及装置、电子设备和存储介质

权利要求书

1.一种场景图生成方法，其特征在于，所述方法包括：

根据图像中各对象的区域框确定对象交互区域，根据对象交互区域和各对象的区域框构建所述图像的全连接图；

将重叠率大于重叠阈值的对象交互区域进行融合，得到所述图像的子图；

根据所述子图和各对象的区域框生成所述图像的场景图。

2.根据权利要求1所述的方法，其特征在于，根据图像中各对象的区域框确定对象交互区域，包括：

根据图像中各对象的区域框和所述区域框的第一置信度，确定对象交互区域和所述对象交互区域的第二置信度，所述第二置信度根据所述第一置信度计算得到；

将重叠率大于重叠阈值的对象交互区域进行融合，得到所述图像的子图，包括：

根据所述第二置信度将重叠率大于重叠阈值的对象交互区域进行融合，得到所述图像的子图。

3.根据权利要求1或2所述的方法，其特征在于，根据所述子图和各对象的区域框生成所述图像的场景图，包括：

将所述子图进行兴趣区域池化处理，得到所述子图的子图特征，将各对象的区域框进行兴趣区域池化处理，得到各对象的对象特征；

根据所述子图的子图特征和各对象的对象特征得到各所述对象之间的关联关系；

根据各所述对象之间的关联关系和各所述对象生成所述图像的场景图。

4.根据权利要求3所述的方法，其特征在于，根据所述子图的子图特征和各对象的对象特征得到各所述对象之间的关联关系，包括：

根据所述子图的子图特征更新各对象的对象特征，得到各对象的对象空间特征；

根据各对象的对象空间特征、所述子图的子图特征得到各对象之间的关联关系。

5.一种场景图生成装置，其特征在于，所述装置包括：

全连接图生成模块，用于根据图像中各对象的区域框确定对象交互区域，根据对象交互区域和各对象的区域框构建所述图像的全连接图；

子图生成模块，用于将重叠率大于重叠阈值的对象交互区域进行融合，得到所述图像的子图；

场景图生成模块，用于根据所述子图和各对象的区域框生成所述图像的场景图。

6.根据权利要求5所述的装置，其特征在于，所述全连接图生成模块，包括：

置信度子模块，用于根据图像中各对象的区域框和所述区域框的第一置信度，确定对象交互区域和所述对象交互区域的第二置信度，所述第二置信度根据所述第一置信度计算得到；

所述子图生成模块，包括：

第一子图生成子模块，用于根据所述第二置信度将重叠率大于重叠阈值的对象交互区域进行融合，得到所述图像的子图。

7.根据权利要求5或6所述的装置，其特征在于，所述场景图生成模块，包括：

特征获取子模块，用于将所述子图进行兴趣区域池化处理，得到所述子图的子图特征，将各对象的区域框进行兴趣区域池化处理，得到各对象的对象特征；

关联关系获取子模块，用于根据所述子图的子图特征和各对象的对象特征得到各所述对象之间的关联关系；

场景图生成子模块，用于根据各所述对象之间的关联关系和各所述对象生成所述图像的场景图。

8.根据权利要求7所述的装置，其特征在于，关联关系获取子模块，包括：

对象空间特征获取单元，用于根据所述子图的子图特征更新各对象的对象特征，得到各对象的对象空间特征；

第一关联关系获取单元，用于根据各对象的对象空间特征、所述子图的子图特征得到各对象之间的关联关系。

9.一种电子设备，其特征在于，包括：处理器；

用于存储处理器可执行指令的存储器；

其中，所述处理器被配置为：执行权利要求1至4中任意一项所述的方法。

10.一种计算机可读存储介质，其上存储有计算机程序指令，其特征在于，所述计算机程序指令被处理器执行时实现权利要求1至4中任意一项所述的方法。