The present invention relates to a device and a vehicle information recognition method, the method comprises: obtaining the training samples, the training sample set including the preset number of training samples; according to the training sample set and the preset optimization goal, the training of convolutional neural network, the optimization objective is to preset the loss function and the loss function corresponding to the corresponding body color models the minimization of summation or convergence; when the training results meet the preset optimization target, parameter information preservation depth convolutional neural network; obtain target image data to identify the vehicle; use the destination image data input through the depth of convolutional neural network parameter information construction, to identify the vehicle models and body color recognition. The public recognition can be carried out simultaneously and two target models of body color, high recognition accuracy; on the two attributes are classified using a convolutional neural network, at the same time improve the recognition rate improves the recognition efficiency, save the running time and memory operation.
【技术实现步骤摘要】
车辆信息识别方法及装置
本公开涉及信息识别
,具体地,涉及一种车辆信息识别方法及装置。
技术介绍
随着现代化经济的高速发展,车辆的数量和种类日益增加,交通监管面临巨大挑战。由于车辆外观复杂多样,受到背景、光照、视角等因素影响,对车辆信息的识别存在很大的难度,准确性难以保证。相关技术中,基于机器学习进行车辆识别时,主要是利用分类器对车辆信息进行单一化目标的识别,例如,仅对车辆的车型进行识别,或者通过学习识别车辆与非车辆的判定边界或验证生成的车辆候选区域等。因此,相关技术中的车辆信息识别,具有车辆参数识别单一化的问题。
技术实现思路
为克服相关技术中存在的问题,本公开提供一种车辆信息识别方法及装置。第一方面,本公开提供一种车辆信息识别方法,包括:获取训练样本集,所述训练样本集中包括预设个数的训练样本,每个训练样本包括:车辆的图像数据、车辆的车型标签和车辆的车身颜色标签;根据所述训练样本集和预设优化目标,对深度卷积神经网络进行训练,所述预设优化目标为车型对应的损失函数和车身颜色对应的损失函数的加权和最小化或收敛;当训练结果满足所述预设优化目标时,保存所述深度卷积神经网络的参数信息;获取包括待识别车辆的目标图像数据;将所述目标图像数据输入通过所述参数信息构建的深度卷积神经网络中,对所述待识别车辆的车型和车身颜色进行识别。在一个实施例中,所述车型对应的损失函数为:所述车身颜色对应的损失函数为:所述车型对应的损失函数和车身颜色对应的损失函数的加权和为:E=λm×Em+(1-λm)×Ec;其中,Em是车型对应的损失函数,Ec是车身颜色对应的损失函数,E所述加权和,zj ...
【技术保护点】
一种车辆信息识别方法,其特征在于,包括:获取训练样本集,所述训练样本集中包括预设个数的训练样本,每个训练样本包括:车辆的图像数据、车辆的车型标签数据和车辆的车身颜色标签数据;根据所述训练样本集和预设优化目标,对深度卷积神经网络进行训练,所述预设优化目标为车型对应的损失函数和车身颜色对应的损失函数的加权和最小化或收敛;当训练结果满足所述预设优化目标时,保存所述深度卷积神经网络的参数信息;获取包括待识别车辆的目标图像数据;将所述目标图像数据输入通过所述参数信息构建的深度卷积神经网络中,对所述待识别车辆的车型和车身颜色进行识别。
【技术特征摘要】
1.一种车辆信息识别方法,其特征在于,包括:获取训练样本集,所述训练样本集中包括预设个数的训练样本,每个训练样本包括:车辆的图像数据、车辆的车型标签数据和车辆的车身颜色标签数据;根据所述训练样本集和预设优化目标,对深度卷积神经网络进行训练,所述预设优化目标为车型对应的损失函数和车身颜色对应的损失函数的加权和最小化或收敛;当训练结果满足所述预设优化目标时,保存所述深度卷积神经网络的参数信息;获取包括待识别车辆的目标图像数据;将所述目标图像数据输入通过所述参数信息构建的深度卷积神经网络中,对所述待识别车辆的车型和车身颜色进行识别。2.根据权利要求1所述的方法,其特征在于,所述车型对应的损失函数为:所述车身颜色对应的损失函数为:所述车型对应的损失函数和车身颜色对应的损失函数的加权和为:E=λm×Em+(1-λm)×Ec;其中,Em是车型对应的损失函数,Ec是车身颜色对应的损失函数,E所述加权和,zj是深度卷积神经网络的全连接层向量j的输出向量,zi是训练样本i的车辆的车型标签和车身颜色标签向量,m是所述训练样本集的车型类别数,c是所述训练样本集的颜色类别数,λm是权重,N是所述训练样本集中的训练样本的数量。3.根据权利要求1所述的方法,其特征在于,所述深度卷积神经网络包括:第一输入层、第二输入层、标签分割层、卷积层、池化层、全连接层、第一输出层和第二输出层;所述对深度卷积神经网络进行训练,以及当训练结果满足所述预设优化目标时,保存所述深度卷积神经网络的参数信息的步骤包括:将每个训练样本的车辆的图像数据通过所述第一输入层输入至所述卷积层;所述第一输入层输入的车辆的图像数据,经过所述卷积层、所述池化层和所述全连接层的逐级变换后,传送至所述第一输出层和所述第二输出层;将每个训练样本的车辆的车型标签数据和车辆的车身颜色标签数据通过所述第二输入层输入至所述标签分割层;在所述标签分割层,对所述第二输入层输入的标签数据进行分割;根据训练样本的车辆的车型标签、第一输出层的输出结果、训练样本的车辆的车身颜色标签和第二输出层的输出结果,调整所述卷积层、所述池化层和所述全连接层的权值和偏置,使得训练结果满足所述预设优化目标;当训练结果满足所述预设优化目标时,分别获取所述卷积层、所述池化层和所述全连接层的权重和偏置;将所述卷积层、所述池化层和所述全连接层的权重和偏置作为所述参数信息进行保存。4.根据权利要求3所述的方法,其特征在于,所述参数信息还包括:所述卷积层的个数、每个卷积层的卷积核大小、所述池化层的个数、每个池化层的大小、所述全连接层的个数以及每个全连接层的大小。5.根据权利要求1所述的方法,其特征在于,所述当训练结果满足所述预设优化目标时,保存所述深度卷积神经网络的参数信息的步骤之后还包括:获取测试样本集,所述测试样本集中包括待测试车辆的车辆图像数据;将测试样本集中的待测试车辆图像数据输入通过所述参数信息构建的深度卷积...
【专利技术属性】
技术研发人员:邹博,邹达,李昕,
申请(专利权)人:东软集团股份有限公司,
类型:发明
国别省市:辽宁,21
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。