[发明专利]车辆识别方法、装置、设备和存储介质

说明书

本发明实施例提供一种车辆识别方法、装置、设备和存储介质，该方法包括：获取包含目标车辆的图像；通过第一模型对图像进行特征提取以得到第一图像特征；通过第二模型对图像进行特征提取以得到第二图像特征，第一模型能够识别的车辆类别是第一类别，第二模型的训练样本图像中所包含车辆属于第二类别，第一类别与第二类别匹配。融合第一图像特征和第二图像特征以得到第三图像特征，根据第三图像特征确定目标对象的类别，以保证图像中目标车辆的类别被准确识别。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种车辆识别方法、装置、设备和存储介质。

背景技术

在很多应用场景中都会遇到图像识别的需求，比如需要识别图像中包含的对象的类别。例如，识别图像中的机动车是什么类型的车辆；识别图像中包含的是什么类别的动物，等等。

目前，为完成上述车辆识别任务，一般做法是：收集训练样本集，基于训练样本集完成车辆识别模型的训练，使用训练得到的车辆识别模型对待识别图像进行车辆识别处理。以机动车分类场景为例，在车辆识别模型训练阶段，会预先收集各类机动车的图片，比如轿车、渣土车、大巴车、搅拌车、公交车，等等，并为收集到的各个图片进行类别标签的标注。

但是实际上，可能存在这样的情况，收集到的某些种类的机动车的图片数量远小于收集到的其他种类的机动车的图片数量。这样会导致训练出的车辆识别模型在样本数量少的机动车类别上的识别能力不佳，最终导致车辆识别准确度不佳。

发明内容

本发明实施例提供一种车辆识别方法、装置、设备和存储介质，用以提高车辆类别识别结果的准确度。

第一方面，本发明实施例提供一种车辆识别方法，所述方法包括：

获取包含目标车辆的图像；

通过第一模型对所述图像进行特征提取，以得到第一图像特征；其中，所述第一模型能够识别的车辆类别包括第一类别；

通过第二模型对所述图像进行特征提取，以得到第二图像特征；其中，所述第二模型的训练样本图像中所包含车辆属于第二类别，所述第一类别与所述第二类别匹配；

融合所述第一图像特征和所述第二图像特征，以得到第三图像特征；

根据所述第三图像特征确定所述目标车辆的类别。

第二方面，本发明实施例提供一种车辆识别装置，所述装置包括：

获取模块，用于获取包含目标车辆的图像；

提取模块，用于通过第一模型对所述图像进行特征提取，以得到第一图像特征；通过第二模型对所述图像进行特征提取，以得到第二图像特征；其中，所述第一模型能够识别的车辆类别包括第一类别，所述第二模型的训练样本图像中所包含车辆属于第二类别，所述第一类别与所述第二类别匹配；

识别模块，用于融合所述第一图像特征和所述第二图像特征，以得到第三图像特征；根据所述第三图像特征确定所述目标车辆的类别。

第三方面，本发明实施例提供一种电子设备，包括：存储器、处理器、通信接口；其中，所述存储器上存储有可执行代码，当所述可执行代码被所述处理器执行时，使所述处理器至少可以实现如第一方面所述的车辆识别方法。

第四方面，本发明实施例提供了一种非暂时性机器可读存储介质，所述非暂时性机器可读存储介质上存储有可执行代码，当所述可执行代码被电子设备的处理器执行时，使所述处理器至少可以实现如第一方面所述的车辆识别方法。

第五方面，本发明实施例提供一种车辆识别方法，所述方法包括：

响应于用户设备调用车辆识别服务接口的请求，利用所述车辆识别服务接口对应的处理资源执行如下步骤：

获取包含目标车辆的图像；