基于双级注意力的图像识别方法及装置

基本信息

申请号 CN202210085821.4 申请日 -
公开(公告)号 CN114386531A 公开(公告)日 2022-04-22
申请公布号 CN114386531A 申请公布日 2022-04-22
分类号 G06K9/62(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I;G06V10/764(2022.01)I;G06V10/774(2022.01)I;G06V10/80(2022.01)I;G06V10/82(2022.01)I 分类 计算;推算;计数;
发明人 张凯;徐卿;袭肖明;杨光远 申请(专利权)人 山东力聚机器人科技股份有限公司
代理机构 北京中和立达知识产权代理有限公司 代理人 祝妍
地址 276808山东省日照市岚山区安东卫街道玉泉二路99号(岚山区科技孵化器电商创业园)
法律状态 -

摘要

摘要 本发明是关于一种基于双级注意力的图像识别方法和装置,其中,方法包括:获取原始图像和对应的二值目标定位图,并对所述原始图像进行预处理,将所述二值目标定位图缩放为预设尺寸;从预处理后的原始图像中提取抽象特征图;根据所述二值目标定位图和抽象特征,计算得到物体特征图和背景特征图;使用空间注意力机制处理所述物体特征图,以得到新的物体特征图;使用通道注意力机制处理所述新的物体特征图和背景特征图融合后的总特征图,以得到新的总特征图;使用全连接层计算所述新的总特征图的特征映射向量,并通过分类器确定分类结果。通过该技术方案,提高网络训练效率,提升预测的准确率。