一种基于注意力机制的特征提取方法、装置及电子设备

基本信息

申请号 CN202010897554.1 申请日 -
公开(公告)号 CN112016571A 公开(公告)日 2020-12-01
申请公布号 CN112016571A 申请公布日 2020-12-01
分类号 G06K9/46(2006.01)I 分类 计算;推算;计数;
发明人 任豪;王紫颐 申请(专利权)人 驭势科技(浙江)有限公司
代理机构 北京开阳星知识产权代理有限公司 代理人 郭鑫
地址 314113 浙江省嘉兴市嘉善县大云镇宏业路299号1号厂房
法律状态 -

摘要

摘要 本公开涉及基于注意力机制的特征提取方法、装置和电子设备。方法包括:获取多个不同尺度的特征图,包括同一深度下尺度最大的第一特征图和至少一个其他尺度的第二特征图;对第一特征图增加注意力;对至少一个第二特征图执行卷积后进行上采样,确定至少一个第三特征图,至少一个第三特征图与第一特征图的尺度相同;基于增加注意力的第一特征图和至少一个第三特征图,确定输出特征图。可见,通过对第一特征图增加注意力,并与其他不同尺度的第二特征图合并,确定输出特征图,使得通过反向传播进行注意力权重参数更新的过程中,可以针对不同的应用场景,通过损失函数约束注意力向应用场景所关心的信息靠近,提升输出特征图在应用场景下使用的准确性。