一种基于注意力机制的图像目标预测方法
基本信息
申请号 | CN202011222985.4 | 申请日 | - |
公开(公告)号 | CN112308081A | 公开(公告)日 | 2021-02-02 |
申请公布号 | CN112308081A | 申请公布日 | 2021-02-02 |
分类号 | G06K9/34;G06K9/62;G06N3/04;G06N3/08 | 分类 | 计算;推算;计数; |
发明人 | 许金泉;王振宁;王溢;蔡碧颖 | 申请(专利权)人 | 南强智视(厦门)科技有限公司 |
代理机构 | 泉州市潭思专利代理事务所(普通合伙) | 代理人 | 廖仲禧;麻艳 |
地址 | 361000 福建省厦门市湖里区火炬高新区软件园创新大厦A区402 | ||
法律状态 | - |
摘要
摘要 | 本发明公开一种基于注意力机制的图像目标预测方法,用于在RGB图像中获取描述语言针对的物体的掩膜;包括如下步骤:提取RGB图像三个尺度的视觉特征;提取描述语言的语言特征;将视觉特征和语言特征进行多模态融合;基于多模态特征以及语言特征,计算四个映射矩阵;然后,通过矩阵运算以及激活函数得到两个注意力图;接着,通过注意力变换计算得到注意力矩阵:最后,将得到的注意力矩阵与原始矩阵进行相加来更新原始矩阵:叠加分组注意力模块,然后通过1x1的卷积神经网络,即得到预测的掩膜。此种方法通过引入受监督的注意力机制,能够增强复杂场景下的推理能力,从而提高检测的精度。 |
