[发明专利]图像识别方法及相关装置

说明书

本申请实施例公开了一种图像识别方法及相关装置，方法包括：获取目标图像，将目标图像输入到主干网络进行特征提取，得到不同尺度下的多个第一特征图，通过多个第一特征图进行后处理，得到至少一个目标，确定多个第一特征图对应的空间注意力特征图，得到多个第一空间注意力特征图，依据多个第一空间注意力特征图确定第一感兴趣区域特征图，将第一感兴趣区域特征图输入到主干网络进行特征提取，得到不同尺度下的多个第二特征图，通过多个第二特征图进行目标属性识别确定至少一个目标属性，将至少一个目标和至少一个目标属性作为目标图像识别结果。采用本申请实施例能够提升图像识别精度和速度。

技术领域

本申请涉及图像处理技术领域，具体涉及一种图像识别方法及相关装置。

背景技术

随着社会的发展，城市汽车数量不断攀升，由此带来的交通和环境问题也日渐加剧。为了应对这些问题，智能交通系统已成为城市发展的重点研究对象。而在智能交通系统中，对于车辆的检索，已经被证明是一项关键技术，在处理包括套牌车、故意遮挡车牌和追踪肇事逃逸等等交通问题上有着得天独厚的优势，对于智能交通系统的建设具有重要的意义，因此，如何提升交通场景的图像识别精度和速度的问题亟待解决。

发明内容

本申请实施例提供了一种图像识别方法及相关装置，可以提升交通场景的图像识别精度和速度。

第一方面，本申请实施例提供一种图像识别方法，应用于电子设备，所述方法包括：

获取目标图像；

将所述目标图像输入到主干网络进行特征提取，得到不同尺度下的多个第一特征图，所述主干网络的每一尺度所在层均为通过通道注意力模块进行自注意力蒸馏学习以及所述主干网络的每一尺度所在层均为通过空间注意力模块进行特征融合提取；

通过所述多个第一特征图进行后处理，得到至少一个目标；

确定所述多个第一特征图对应的空间注意力特征图，得到多个第一空间注意力特征图；

依据所述多个第一空间注意力特征图确定第一感兴趣区域特征图；

将所述第一感兴趣区域特征图输入到所述主干网络进行特征提取，得到不同尺度下的多个第二特征图；

通过所述多个第二特征图进行目标属性识别确定至少一个目标属性；

将所述至少一个目标和所述至少一个目标属性作为目标图像识别结果。

第二方面，本申请实施例提供了一种图像识别装置，应用于电子设备，所述装置包括：获取单元、提取单元、处理单元和确定单元，其中，

所述获取单元，用于获取目标图像；

所述提取单元，用于将所述目标图像输入到主干网络进行特征提取，得到不同尺度下的多个第一特征图，所述主干网络的每一尺度所在层均为通过通道注意力模块进行自注意力蒸馏学习以及所述主干网络的每一尺度所在层均为通过空间注意力模块进行特征融合提取；

所述处理单元，用于通过所述多个第一特征图进行后处理，得到至少一个目标；

所述确定单元，用于确定所述多个第一特征图对应的空间注意力特征图，得到多个第一空间注意力特征图；

所述确定单元，还用于依据所述多个第一空间注意力特征图确定第一感兴趣区域特征图；

所述提取单元，还用于将所述第一感兴趣区域特征图输入到所述主干网络进行特征提取，得到不同尺度下的多个第二特征图；

所述确定单元，还用于通过所述多个第二特征图进行目标属性识别确定至少一个目标属性；以及将所述至少一个目标和所述至少一个目标属性作为目标图像识别结果。