【技术实现步骤摘要】
一种图像识别方法、装置、电子设备及存储介质
[0001]本申请涉及图像处理
,特别涉及一种图像识别方法、装置、电子设备及存储介质。
技术介绍
[0002]在图像识别场景中,如何进行图像识别,以保证图像识别的准确率及计算量的平衡成为问题。
技术实现思路
[0003]本申请提供如下技术方案:
[0004]本申请一方面提供一种图像识别方法,包括:
[0005]基于待识别图像对应的第一特征图,获得至少一种尺度的第二特征图;
[0006]基于所述至少一种尺度中各种所述尺度的第二特征图,确定各种所述尺度对应的分层系数,所述分层系数用于表征识别模块的多个识别层中各个所述识别层对所述尺度的第二特征图进行图像识别的权重,所述识别层用于对所述尺度的第二特征图进行图像识别;
[0007]基于所述分层系数,从所述识别模块的多个识别层中选择待使用层,以基于所述待使用层对所述尺度的第二特征图进行图像识别。
[0008]所述图像识别方法基于多任务自主分层模型执行,所述多任务自主分层模型包括:特 ...
【技术保护点】
【技术特征摘要】
1.一种图像识别方法,包括:基于待识别图像对应的第一特征图,获得至少一种尺度的第二特征图;基于所述至少一种尺度中各种所述尺度的第二特征图,确定各种所述尺度对应的分层系数,所述分层系数用于表征识别模块的多个识别层中各个所述识别层对所述尺度的第二特征图进行图像识别的权重,所述识别层用于对所述尺度的第二特征图进行图像识别;基于所述分层系数,从所述识别模块的多个识别层中选择待使用层,以基于所述待使用层对所述尺度的第二特征图进行图像识别。2.根据权利要求1所述的方法,所述图像识别方法基于多任务自主分层模型执行,所述多任务自主分层模型包括:特征提取模块、分层估计模块和识别模块;所述特征提取模块,用于提取待识别图像对应的第一特征图;所述分层估计模块,用于基于所述待识别图像对应的第一特征图,获得至少一种尺度的第二特征图,基于所述至少一种尺度中各种所述尺度的第二特征图,确定各种所述尺度对应的分层系数;所述识别模块,用于基于所述分层系数,从所述识别模块的多个识别层中选择待使用层,以基于所述待使用层对所述尺度的第二特征图进行图像识别。3.根据权利要求2所述的方法,所述分层估计模块包括:第一提取模块和系数确定模块;所述第一提取模块,用于提取待识别图像对应的第一特征图对应的至少一种尺度的候选框;以及,基于所述至少一种尺度的候选框和所述第一特征图,确定得到至少一种尺度的第二特征图;所述系数确定模块,用于基于所述至少一种尺度中各种所述尺度的第二特征图,确定各种所述尺度对应的识别模块的多个识别层中各个所述识别层能够识别出所述尺度的第二特征图所占的第一概率,所述第一概率表征所述识别层对所述尺度的第二特征图进行图像识别的权重。4.根据权利要求3所述的方法,所述系数确定模块包括:第二提取模块、第三提取模块和分类模块;基于所述至少一种尺度中各种所述尺度的第二特征图,确定各种所述尺度对应的识别模块的多个识别层中各个所述识别层能够识别出所述尺度的第二特征图所占的第一概率,包括:将所述至少一种尺度中各种所述尺度的第二特征图输入所述第二提取模块,得到所述第二提取模块确定的各种所述尺度的第三特征图;将各种所述尺度的第三特征图输入所述第三提取模块,得到所述第三提取模块确定的各种所述尺度的第四特征图;将各种所述尺度的第四特征图输入所述分类模块,得到所述分类模块确定的各种所述尺度对应的识别模块的多个识别层中各个所述识别层能够识别出所述尺度的第二特征图所占的第一概率。5.根据权利要求1所述的方法,基于所述分层系数,从所述识别模块的多个识别层中选择待使用层,包括:
基于所述分层系数,从所述识别模块的多个识别层中选择权重排列在前n个的识别层为待使用层;或,基于所述分层系数,从所述识别模块的多个识别层中选择权重最大的识别层为待使用层。6.根据权利要求3所述的方法,所述第一提取模块、所述系数确定模块和所述识别模块通过以下方式进行训练:获取至少一个任务对应的样本图像对应的第一样本特征图;基于第一提取模块提取所述样本图像对应的第一样本特征图对应的至少一种尺度的第二样本特征图;将所述至少一种尺度中各种所述尺度的第二样本特征图输入所述系数确定模块,得到所述系数确定模块确...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。