基于注意力机制的目标检测方法、网络、设备和存储介质

基本信息

申请号 CN202010727998.0 申请日 -
公开(公告)号 CN112016569A 公开(公告)日 2020-12-01
申请公布号 CN112016569A 申请公布日 2020-12-01
分类号 G06K9/46;G06K9/32;G06K9/62 分类 计算;推算;计数;
发明人 任豪;郭钰 申请(专利权)人 驭势科技(浙江)有限公司
代理机构 北京开阳星知识产权代理有限公司 代理人 郭鑫
地址 211100 江苏省南京市江宁开发区苏源大道19号九龙湖国际企业总部园B4座二层(江宁开发区)
法律状态 -

摘要

摘要 本申请涉及图像处理技术领域,公开了一种基于注意力机制的目标检测方法、网络、设备和存储介质。该方法包括:通过特征提取网络提取待检测图像的特征,得到第一特征图;通过区域生成网络处理第一特征图,得到至少一个候选区域;通过ROI池化网络处理第一特征图和候选区域,得到第二特征图;通过分类网络和回归网络分别处理第二特征图,得到至少一个目标的类别得分和边界框信息;特征提取网络、分类网络和回归网络中的至少一个包括注意力子网络;注意力子网络确定在先层和临时层;基于在先层和临时层,确定注意力权重;将在先层和注意力权重进行乘积操作,对在先层增加注意力。通过上述技术方案,提高目标检测的速度和精确度。