基于语义空间约束和注意力机制的场景图生成方法

基本信息

申请号 CN201910250400.0 申请日 -
公开(公告)号 CN110084128B 公开(公告)日 2021-12-14
申请公布号 CN110084128B 申请公布日 2021-12-14
分类号 G06K9/00(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 汪从玲 申请(专利权)人 安徽省科亿信息科技有限公司
代理机构 合肥市上嘉专利代理事务所(普通合伙) 代理人 李璐;郭华俊
地址 230000 安徽省合肥市高新区黄山路602号国家大学科技园创业孵化中心
法律状态 -

摘要

摘要 本发明公开了一种基于语义空间约束和注意力机制的场景图生成方法,包括以下步骤:S1:采用VG数据集,随机选取若干张图像作为验证集,在剩余图像中按比例随机选取,分别作为训练集和测试集;S2:基于深度学习的方法和VG数据集,构建目标检测网络;S3:建立关系推理网络;S4:根据关系推理网络得到的语义约束信息和注意力网络,得到突出可能存在相关关系的区域和通道特征,进行关系推理分类。本发明无需使用图相关信息流传递机制,基于语义空间的约束和注意力机制推测两两物体之间的相关关系,没有引入更多的噪声,网络比较简单,提供了更高的关系推理精度,能够更有效提取出存在物体相关关系的特征。