[发明专利]用于识别车辆的方法和装置

说明书

本申请实施例公开了用于识别车辆的方法和装置。该方法的一具体实施方式包括：获取待识别图像，其中，待识别图像包括待识别车辆图像区域；确定待识别车辆图像区域在待识别图像中的位置；基于待识别车辆图像区域在待识别图像中的位置，从待识别图像中分割出待识别车辆图像；将待识别车辆图像输入至预先训练的识别模型，得到待识别车辆图像的识别结果，其中，识别结果包括待识别车辆属于预设种车辆型号中的每种车辆型号的置信度，识别模型用于识别车辆所属的车辆型号。该实施方式利用用于识别车辆所属的车辆型号的识别模型识别车辆型号，提高了车辆型号的识别准确度。

技术领域

本申请实施例涉及计算机技术领域，具体涉及图像识别技术领域，尤其涉及用于识别车辆的方法和装置。

背景技术

随着车辆型号的不断增多，人们通过肉眼通常只能辨别出极少数型号的车辆。因此，帮助用户进行车辆识别成为了一种需求。现有的车辆识别方式通常是通过相似图进行检索查询来识别图像中的车辆的型号。

发明内容

本申请实施例提出了用于识别车辆的方法和装置。

第一方面，本申请实施例提供了一种用于识别车辆的方法，该方法包括：获取待识别图像，其中，待识别图像包括待识别车辆图像区域；确定待识别车辆图像区域在待识别图像中的位置；基于待识别车辆图像区域在待识别图像中的位置，从待识别图像中分割出待识别车辆图像；将待识别车辆图像输入至预先训练的识别模型，得到待识别车辆图像的识别结果，其中，识别结果包括待识别车辆属于预设种车辆型号中的每种车辆型号的置信度，识别模型用于识别车辆所属的车辆型号。

在一些实施例中，确定待识别车辆图像区域在待识别图像中的位置，包括：将待识别图像输入至预先训练的检测模型，得到待识别车辆图像区域在待识别图像中的位置，其中，检测模型用于检测车辆图像区域在图像中的位置。

在一些实施例中，在确定待识别车辆图像区域在待识别图像中的位置之前，还包括：对待识别图像进行旋转，得到多张不同角度的待识别图像；基于待识别车辆图像区域在待识别图像中的位置，从待识别图像中分割出待识别车辆图像，包括：基于待识别车辆图像区域在多张不同角度的待识别图像中的位置，从多张不同角度的待识别图像中选取出待识别图像，并从所选取出的待识别图像中分割出待识别车辆图像。

在一些实施例中，该方法还包括：按照数值大小顺序对识别结果中的待识别车辆属于预设种车辆型号中的每种车辆型号的置信度进行排序；从数值大的一侧开始选取出预设数目个置信度，输出所选取的置信度和与所选取出的置信度对应的车辆型号。

在一些实施例中，该方法还包括：将识别结果中的待识别车辆属于预设种车辆型号中的每种车辆型号的置信度与预设阈值进行比较；若存在大于预设阈值的置信度，输出大于预设阈值的置信度和与大于预设阈值的置信度对应的车辆型号。

在一些实施例中，检测模型通过如下步骤训练得到：获取训练样本集，其中，训练样本集中的每个训练样本包括样本图像，样本图像包括样本车辆图像区域；对于训练样本集中的每个训练样本，执行以下操作：对该训练样本中的样本图像中的样本车辆图像区域进行位置标注，得到该训练样本中的样本图像中的样本车辆图像区域的位置标注结果；将该训练样本中的样本图像输入至检测模型，得到样本车辆图像区域在该训练样本中的样本图像中的位置；确定样本车辆图像区域在该训练样本中的样本图像中的位置和该训练样本中的样本图像中的样本车辆图像区域的位置标注结果之间的第一差异；基于第一差异调整检测模型的参数。