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

    公开(公告)号:CN114386531A

    公开(公告)日:2022-04-22

    申请号:CN202210085821.4

    申请日:2022-01-25

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

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

    公开(公告)号:CN114386531B

    公开(公告)日:2023-02-14

    申请号:CN202210085821.4

    申请日:2022-01-25

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