基于注意力机制的目标检测方法、网络、设备和存储介质
基本信息
申请号 | CN202010727998.0 | 申请日 | - |
公开(公告)号 | CN112016569A | 公开(公告)日 | 2020-12-01 |
申请公布号 | CN112016569A | 申请公布日 | 2020-12-01 |
分类号 | G06K9/46;G06K9/32;G06K9/62 | 分类 | 计算;推算;计数; |
发明人 | 任豪;郭钰 | 申请(专利权)人 | 驭势科技(浙江)有限公司 |
代理机构 | 北京开阳星知识产权代理有限公司 | 代理人 | 郭鑫 |
地址 | 211100 江苏省南京市江宁开发区苏源大道19号九龙湖国际企业总部园B4座二层(江宁开发区) | ||
法律状态 | - |
摘要
摘要 | 本申请涉及图像处理技术领域,公开了一种基于注意力机制的目标检测方法、网络、设备和存储介质。该方法包括:通过特征提取网络提取待检测图像的特征,得到第一特征图;通过区域生成网络处理第一特征图,得到至少一个候选区域;通过ROI池化网络处理第一特征图和候选区域,得到第二特征图;通过分类网络和回归网络分别处理第二特征图,得到至少一个目标的类别得分和边界框信息;特征提取网络、分类网络和回归网络中的至少一个包括注意力子网络;注意力子网络确定在先层和临时层;基于在先层和临时层,确定注意力权重;将在先层和注意力权重进行乘积操作,对在先层增加注意力。通过上述技术方案,提高目标检测的速度和精确度。 |
