[发明专利]基于注意力模型的图像识别方法和系统在审
申请号: | 201810139775.5 | 申请日: | 2018-02-11 |
公开(公告)号: | CN108364023A | 公开(公告)日: | 2018-08-03 |
发明(设计)人: | 张志伟;杨帆 | 申请(专利权)人: | 北京达佳互联信息技术有限公司 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06N3/04 |
代理公司: | 北京市立方律师事务所 11330 | 代理人: | 刘延喜 |
地址: | 100084 北京市海淀区*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明提供一种基于注意力模型的图像识别方法和系统,首先获取图像矩阵形状为[W,H,C]的输入特征图,其中,W为宽度,H为高度,C为通道数;然后使用预设的空间映射权重矩阵对输入特征图进行空间映射,并经过激活函数激活后得到空间权重矩阵,将所述空间权重矩阵与输入特征图的图像矩阵按位相乘得到输出特征图,其中,所述预设的空间映射权重矩阵为注意力在于图像宽度和高度的空间注意力矩阵[C,1],此时空间权重矩阵的形状为[W,H,1],或者所述预设的空间映射权重矩阵为注意力在于图像通道数的通道注意力矩阵[C,C],此时空间权重矩阵的形状为[1,1,C],可有效提高特征提取的针对性,从而强化对于图像局部特征的提取能力。 | ||
搜索关键词: | 矩阵 空间映射 权重矩阵 输入特征 注意力 空间权 预设 注意力模型 图像识别 图像局部特征 相乘 获取图像 激活函数 矩阵形状 输出特征 所述空间 特征提取 图像矩阵 图像通道 通道数 激活 图像 | ||
【主权项】:
1.一种基于注意力模型的图像识别方法,其特征在于,包括如下步骤:步骤S10:获取图像矩阵形状为[W,H,C]的输入特征图,其中,W为宽度,H为高度,C为通道数;步骤S20:使用预设的空间映射权重矩阵对输入特征图进行空间映射,并经过激活函数激活后得到空间权重矩阵,将所述空间权重矩阵与输入特征图的图像矩阵按位相乘得到输出特征图,其中,所述预设的空间映射权重矩阵为注意力在于图像宽度和高度的空间注意力矩阵[C,1],此时空间权重矩阵的形状为[W,H,1],或者所述预设的空间映射权重矩阵为注意力在于图像通道数的通道注意力矩阵[C,C],此时空间权重矩阵的形状为[1,1,C]。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京达佳互联信息技术有限公司,未经北京达佳互联信息技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201810139775.5/,转载请声明来源钻瓜专利网。