CN107665353A - 基于卷积神经网络的车型识别方法、装置、设备及计算机可读存储介质 - Google Patents

Abstract

本发明实施例公开了一种基于卷积神经网络的车型识别方法、装置、设备及计算机可读存储介质，其中所述方法包括对获取的待测图片进行预处理；将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；确定所有概率值中的最大概率值，并将最大概率值对应的车型作为所述待测图片的车型；所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。本发明能够实现高精度的车型识别，同时使得识别过程高效稳定。

Description

基于卷积神经网络的车型识别方法、装置、设备及计算机可读 存储介质

技术领域

本发明涉及计算机技术领域，尤其涉及一种基于卷积神经网络的车型识别方法、装置、设备及计算机可读存储介质。

背景技术

汽车车型识别在车辆管理、车辆违规逃逸、车辆巡查布控和事故车损赔付等诸多问题上都起到关键作用，车辆类型具有不易改变的优点，成为车辆辨识中非常重要的特征，而在无牌车、套牌车等情况下，无法通过车牌识别和图像清晰化处理技术获取有效的车辆信息，特别是在车损理赔时，车型对赔付金额影响巨大，车型识别在其他类似交通监控和管制、交通事故责任认定等诸多场景中也起着非常重要的作用。

传统的车型识别方法，往往只能针对不同的任务设计诸如基于尺度不变特征转换(SIFT)等不同的特征，然后利用支持向量机(SVM)或随机森林(Random Forest)等分类器进行训练，但往往只能应用到车辆种类和车牌识别的场景中，对于多达上千种且很多类型非常相似的车型来说，很难人工设计针对性的特征进行识别。另外虽然还有基于多传感器融合的车型识别方法，但是该方法的原理和识别非常简单，存在着对外界环境敏感、故障率较高的缺点。

发明内容

本发明实施例提供一种基于卷积神经网络的车型识别方法、装置、设备及计算机可读存储介质，能够实现高精度的车型识别，同时使得识别过程高效稳定。

一方面，本发明实施例提供了一种基于卷积神经网络的车型识别方法，该方法包括：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。

进一步地，所述预设的图片数据包括预设的第一图片数据，所述对获取的待测图片进行预处理之前，包括：

将预设的第一图片数据分为第一训练集以及第一验证集；

利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型；

利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本；

若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型；

利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

进一步地，所述预设的图片数据还包括预设的第二图片数据，所述方法还包括：

将预设的第二图片数据分为第二训练集以及第二验证集；

利用所述第二训练集对第二卷积神经网络进行训练，以得到对应的第二中间模型；

利用所述第二验证集对所述第二中间模型进行验证，以得到对应的第二错误集，其中所述第二错误集包括至少一个第二错误样本；

若所述第二错误集中第二错误样本的数量大于或等于第二预设阀值，利用所述第二错误集训练所述第二中间模型以得到一个新的第二中间模型；

利用所述第二验证集对所述新的第二中间模型再次进行验证，直至所述第二错误集中第二错误样本的数量小于预设阀值，并判定此时的新的第二中间模型为对应的第二预设检测模型。

进一步地，所述第一卷积神经网络包括八层结构，所述第二卷积神经网络包括二十层结构，其中，所述第一卷积神经网络包括五个卷积层、两个全连接层以及一个用于二分类的概率统计层。

进一步地，所述将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息，包括：

将预处理后的待测图片输入第一预设检测模型以得到一置信度；

判断所述置信度是否大于预设临界值；

其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

另一方面，本发明实施例还提供了一种基于卷积神经网络的车型识别装置，该装置包括：

处理单元，用于对获取的待测图片进行预处理；

判断单元，用于将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

输入单元，用于若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

计算单元，用于通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定单元，用于确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。

进一步地，所述预设的图片数据包括预设的第一图片数据，所述装置还包括：

分类单元，用于将预设的第一图片数据分为第一训练集以及第一验证集；

训练单元，用于利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型；

验证单元，用于利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本；

调整单元，用于若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型；

判定单元，用于利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

进一步地，所述判断单元包括：

处理单元，用于对获取的待测图片进行预处理；

置信度获取单元，用于将预处理后的待测图片输入第一预设检测模型以得到一置信度；

置信度判断单元，用于判断所述置信度是否大于预设临界值；

其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

又一方面，本发明实施例还提供了一种基于卷积神经网络的车型识别设备，该设备包括：

存储器，用于存储实现车型识别的程序；以及

处理器，用于运行所述存储器中存储的实现车型识别的程序，以执行以下操作：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。

再一方面，本发明实施例还提供了一种计算机可读存储介质，所述计算机可读存储介质存储有一个或者一个以上程序，所述一个或者一个以上程序可被一个或者一个以上的处理器执行，以实现以下步骤：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。

本发明实施例通过将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；具体的是先对待测图片进行车辆二分类，再根据车辆二分类结果找到含有车辆特征信息的图片，并再次进行车型的分类识别，从而使得该发明实施例能够实现对车辆的精细车型的分类识别，例如实现高达92.48％的高精度的车型识别，同时也能够使得识别过程更为高效稳定。

附图说明

为了更清楚地说明本发明实施例技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图是本发明的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。

图1是本发明实施例提供的基于卷积神经网络的车型识别方法的示意流程图；

图2是本发明实施例提供的基于卷积神经网络的车型识别方法的另一示意流程图；

图3是本发明实施例提供的基于卷积神经网络的车型识别方法的另一示意流程图；

图4是本发明实施例提供的基于卷积神经网络的车型识别方法的另一演示示意图；

图5是本发明实施例提供的基于卷积神经网络的车型识别装置的示意性框图；

图6是本发明实施例提供的基于卷积神经网络的车型识别装置的另一示意性框图；

图7是本发明实施例提供的基于卷积神经网络的车型识别装置的另一示意性框图；

图8是本发明实施例提供的基于卷积神经网络的车型识别设备的结构组成示意图。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。

应当理解，当在本说明书和所附权利要求书中使用时，术语“包括”和“包含”指示所描述特征、整体、步骤、操作、元素和/或组件的存在，但并不排除一个或多个其它特征、整体、步骤、操作、元素、组件和/或其集合的存在或添加。

还应当理解，在此本发明说明书中所使用的术语仅仅是出于描述特定实施例的目的而并不意在限制本发明。如在本发明说明书和所附权利要求书中所使用的那样，除非上下文清楚地指明其它情况，否则单数形式的“一”、“一个”及“该”意在包括复数形式。

请参阅图1，图1是本发明实施例提供的一种基于卷积神经网络的车型识别方法的示意流程图。该方法可以运行在智能手机(如Android手机、IOS手机等)、平板电脑、笔记本电脑以及智能设备等设备中。本发明的方法可以自动分析输入的待测图片，从而实现对车辆的精细车型的分类识别，例如实现高达92.48％的高精度的车型识别，同时也能够使得识别过程更为高效稳定。如图1所示，该方法包括步骤S101～S105。

S101对获取的待测图片进行预处理。

在本发明实施例中，获取待测图片后，需要对其进行一定的处理。待测图片可以是常规的图片，也可以是从视频数据中提取视频关键帧从而得到的图片。为了提高分类检测的准确性，需要对待测图片进行相应的处理。尤其是，需要对待测图片进行特征强化，即统一化到227*227大小后的图像。

S102，将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息。

在本发明实施例中，需要将预处理后的待测图片输入第一预设检测模型进行二分类，从而判定所述待测图片是否含有车辆特征信息。其中，第一预设检测模型可以是一个基于深度卷积神经网路和大量车辆相关图片数据集训练出来的车辆二分类模型。

具体的，如图2所示，作为优选的实施例，步骤S102包括步骤S201～S202，

S201，将预处理后的待测图片输入第一预设检测模型以得到一置信度。

在本发明实施例中，将已进行预处理的待测图片输入第一预设检测模型后，可以得到一个相应的置信度，因所述第一预设检测模型是基于深度卷积神经网络而训练得到的，故该置信度可以由该深度卷积神经网络中的用于二分类的连接层作为输出。

S202，判断所述置信度是否大于预设临界值。

在本发明实施例中，该置信度如果不大于预设置信度，那么可知，该待测图片不含有车辆特征信息，那么也即图片不含有车辆，不需要进行后续的车型识别。其中，所述预设置信度可以根据实际情况进行相应的设定。比如，当预设置信度为0.6时，如果所述置信度小于或等于0.6，那么该待测图片为含有车辆特征信息的图片。即，其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

S103，若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型。

其中，在本发明实施例中，当所述待测图片含有车辆特征信息时，要将预处理后的待测图片输入第二预设检测模型，所述第二预设检测模型可以是一个基于深度卷积神经网路和大量车辆相关图片数据集训练出来的车型分类模型。

S104，通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值。

在本发明实施例中，所述进行处理后的待测图片输入所述第二预设检测模型后可以计算得出所述待测图片对应的每一类车型的概率值。其中，本发明实施例中的车型可以包括品牌名、厂商名以及车系型号等信息，当然也不仅限于上述信息。一般情况下，可以通过第二预设检测模型计算输出的每个概率值均对应一个车型。

例如，所述车型可以包括如下面表1所示的车型，

序号	品牌名	厂商名	车系型号
				1	路虎	路虎(进口)	揽胜
2	路虎	路虎(进口)	第三代发现
				3	路虎	路虎(进口)	神行者2
4	路虎	路虎(进口)	揽胜极光(进口)
				5	路虎	路虎(进口)	第四代发现
6	路虎	路虎(进口)	揽胜运动版
				7	路虎	路虎(进口)	seriesI
8	路虎	路虎(进口)	神行者
				9	路虎	路虎(进口)	第一代发现
10	路虎	路虎(进口)	第二代发现
				11	路虎	路虎(进口)	揽胜极光
12	路虎	路虎(进口)	发现神行

表1

其中，序号1到序号12中的每一个车型都对应有一个概率，假若序号9所表示的车型对应的概率最大，那么此时所述待测图片中的车辆的车型即为品牌名为路虎，厂商名为路虎(进口)，车系型号为第一代发现的型号。

S105，确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型。

在本发明实施例中，对所有的概率值进行对比分析，从而获取最大概率值，此时，该概率值所对应有的车型即为所述待测图片的车型。

另外，作为优选的实施例，如图3所示，所述预设的图片数据包括预设的第一图片数据，本发明实施例提供的一种基于卷积神经网络的车型识别方法中的步骤S101之前还包括步骤S301～S305：

S301，将预设的第一图片数据分为第一训练集以及第一验证集。

在本发明实施例中，预设的第一图片数据可以包括预设分类的包括车辆的图片和不包括车辆的图片，其可由人工进行分类筛选，将这两类数据作为两个场景，并作为输入数据提供给卷积神经网络进行学习分类，从而得到可以判断图片是否为含有车辆的图片的车辆二分类模型。

具体的，可以对不含有车辆的图片给定分类标注为0，对含有车辆的图片给定分类标注为1，再对已进行标注的预设的图片数据按4:1的比例分配为第一训练集以及第一验证集，第一训练集中可以包括含有车辆的图片和不含有车辆的图片，第一验证集中可以包括含有车辆的图片和不含有车辆的图片。

其中，第一训练集用于对卷积神经网络进行常规训练，而第一验证集用于对训练后的卷积神经网络得到的模型进行相应的分类检测，为了提高识别的准确度，可以对第一训练集和第一验证集先进行相应的预处理，比如特征强化等，然后才能输入卷积神经网络进行训练。

S302，利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型。

在本发明实施例中，卷积神经网络(Convolutional Neural Network，CNN)是一种前馈神经网络，它的人工神经元可以响应一部分覆盖范围内的周围单元，对于大型图像处理有出色表现。不同的卷积神经网络包括不同的分层结构。具体的，本发明实施例可以通过第一训练集对选用的深度卷积神经网路来训练得到第一中间模型。

作为优选的实施例，所述第一卷积神经网络可以包括八层结构，其中，所述第一卷积神经网络包括五个卷积层、两个全连接层以及一个用于二分类的概率统计层。其中前五层为卷积层，用于进行特征提取和降维，后两层为全连接层，最后为用于二分类的概率统计层。第一卷积神经网络中的每层卷积层可以通过卷积核将输入的图片数据过滤为二维向量，在训练阶段对其参数进行单独计算，而全连接层将输入和带权重的向量进行点乘，因此后一层的神经元与前一层的神经元全部连接起来，所有神经元都通过激活函数进行学习加速，而概率统计层用于进行图片是否包括车辆特征信息的判断。

S303，利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本。

在本发明实施例中，将第一验证集中的图片输入第一中间模型中进行分类检测以得到分类检测结果，当分类检测结果与该图片的预设分类不一致时，将该图片认定为一个第一错误样本。具体可以将所有分类检测结果与图片的预分类不一致的图片归类为第一错误集。综上可知，每个第一错误集均可以包括至少一个第一错误样本。

S304，若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型。

在本发明实施例中，如果第一错误集中错误样本的数量大于或等于预设阈值，则表明此时分类检测的结果的错误率在不可接受的范围内，此时需要利用所述第一错误集训练所述第一中间模型以得到一个对应的新的第一中间模型，从而进一步提高第一中间模型的分类检测的准确度。

S305，利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

在本发明实施例中，得到新的第一中间模型后，需要再利用第一验证集再次进行验证以得到新的分类检测结果，同时判断此时的第一错误集中的错误样本的数量是否小于预设阀值，当第一错误集中的错误样本的数量小于预设阀值，那么则可以判定此时的新的第一中间模型为对应的第一预设检测模型。而如果第一错误集中的错误样本的数量大于预或等于预设阀值，那么则可以返回步骤S304。

故通过第一预设检测模型对处理后的待测图片进行二分类的时候，以通过第一层卷积层为例，输入统一化到227*227大小后的图片，卷积核大小为11*11，步长为4，卷积核个数为96，卷积后的特征图减去边缘后大小为55，特征图经过ReLu激活函数、Norm归一化后通过池化操作降维，最终输出大小为27*27*96的特征图，然后再输入后续的卷积层以及全连接层中以进行二分类。

作为优选的实施例，如图4所示，所述预设的图片数据还包括预设的第二图片数据，本发明实施例提供的一种基于卷积神经网络的车型识别方法中的步骤S101之前还可以包括步骤S401～S405：

S401，将预设的第二图片数据分为第二训练集以及第二验证集。

在本发明实施例中，预设的第二图片数据可以包括预设分类的具有各种车型的车辆的图片，其可由人工进行分类筛选对应车型，将每张图片作为一个场景，并作为输入数据提供给一个卷积神经网络进行学习分类，从而得到可以判断图片中的车辆车型的车型分类模型。

其中，第二训练集用于对卷积神经网络进行常规训练，而第二验证集用于对训练后的卷积神经网络得到的模型进行相应的分类检测，为了提高识别的准确度，可以对第二训练集和第二验证集先进行相应的预处理，比如特征强化等，然后才能输入卷积神经网络进行训练。

S402，利用所述第二训练集对第二卷积神经网络进行训练，以得到对应的第二中间模型。

在本发明实施例中，卷积神经网络(Convolutional Neural Network，CNN)是一种前馈神经网络，它的人工神经元可以响应一部分覆盖范围内的周围单元，对于大型图像处理有出色表现。不同的卷积神经网络包括不同的分层结构。具体的，本发明实施例可以通过第二训练集对选用的深度卷积神经网路来训练得到第二中间模型。

S403，利用所述第二验证集对所述第二中间模型进行验证，以得到对应的第二错误集，其中所述第二错误集包括至少一个第二错误样本。

在本发明实施例中，将第二验证集中的图片输入第二中间模型中进行分类检测以得到分类检测结果，当分类检测结果与该图片的预设分类不一致时，将该图片认定为一个第二错误样本。具体可以将所有分类检测结果与图片的预分类不一致的图片归类为第二错误集。综上可知，每个第二错误集均可以包括至少一个第二错误样本。

S404，若所述第二错误集中第二错误样本的数量大于或等于第二预设阀值，利用所述第二错误集训练所述第二中间模型以得到一个新的第二中间模型。

在本发明实施例中，如果第二错误集中错误样本的数量大于或等于预设阈值，则表明此时分类检测的结果的错误率在不可接受的范围内，此时需要利用所述第二错误集训练所述第二中间模型以得到一个对应的新的第二中间模型，从而进一步提高第二中间模型的分类检测的准确度。

S405，利用所述第二验证集对所述新的第二中间模型再次进行验证，直至所述第二错误集中第二错误样本的数量小于预设阀值，并判定此时的新的第二中间模型为对应的第二预设检测模型。

在本发明实施例中，得到新的第二中间模型后，需要再利用第二验证集再次进行验证以得到新的分类检测结果，同时判断此时的第二错误集中的错误样本的数量是否小于预设阀值，当第二错误集中的错误样本的数量小于预设阀值，那么则可以判定此时的新的第二中间模型为对应的第二预设检测模型。而如果第二错误集中的错误样本的数量大于预或等于预设阀值，那么则可以返回步骤S404。

其中，第二卷积神经网路可以包括二十层结构，如表2所示。

表2

该第二卷积神经网络使用了大量的1*1的卷积核，增强了对非线性的拟合用于降维，同时加入了初始模块(Inception)，使用不同尺度的滤波器来解决尺度问题。同时其余各层都是类似，只是对滤波器的个数做了改动。另外，该第二卷积神经网络可以包括三个输出层，其中最后一个输出层的结构最优，故事利用最后一层的输出层得出的结果作为输出。故通过对该第二卷积神经网络对数十万张确定车型的车辆图片进行训练，可以生成保有车型特征参数的车型分类模型。

由以上可见，本发明实施例通过将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；具体的是先对待测图片进行车辆二分类，再根据车辆二分类结果找到含有车辆特征信息的图片，并再次进行车型的分类识别，从而使得该发明实施例能够实现对车辆的精细车型的分类识别，即实现高达92.48％的高精度的车型识别，同时也能够使得识别过程更为高效稳定。

请参阅图5，对应上述一种基于卷积神经网络的车型识别方法，本发明实施例还提出一种基于卷积神经网络的车型识别装置，该装置100包括：处理单元101、判断单元102、输入单元103、计算单元104、确定单元105。

其中，所述处理单元101，用于对获取的待测图片进行预处理。

所述判断单元102用于将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息。

输入单元103用于若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型。

计算单元104用于通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值。

确定单元105用于确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型。

如图6所示，作为优选的实施例，所述预设的图片数据包括预设的第一图片数据，所述装置100还包括分类单元201、训练单元202、验证单元203、调整单元204、判定单元205。

其中，分类单元201用于将预设的第一图片数据分为第一训练集以及第一验证集；

训练单元202用于利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型；

验证单元203用于利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本；

调整单元204用于若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型；

判定单元205用于利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

作为优选的，所述预设的图片数据还包括预设的第二图片数据，其中分类单元201还用于将预设的第二图片数据分为第二训练集以及第二验证集。

训练单元202还用于利用所述第二训练集对第二卷积神经网络进行训练，以得到对应的第二中间模型。

验证单元203还用于利用所述第二验证集对所述第二中间模型进行验证，以得到对应的第二错误集，其中所述第二错误集包括至少一个第二错误样本。

调整单元204还用于若所述第二错误集中第二错误样本的数量大于或等于第二预设阀值，利用所述第二错误集训练所述第二中间模型以得到一个新的第二中间模型。

判定单元205还用于利用所述第二验证集对所述新的第二中间模型再次进行验证，直至所述第二错误集中第二错误样本的数量小于预设阀值，并判定此时的新的第二中间模型为对应的第二预设检测模型。

如图7所示，所述处理单元101，包括置信度获取单元301、置信度判断单元302。

其中，置信度获取单元301用于将预处理后的待测图片输入第一预设检测模型以得到一置信度。

置信度判断单元302用于判断所述置信度是否大于预设临界值。

其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

上述基于卷积神经网络的车型识别装置可以实现为一种计算机程序的形式，该计算机程序可以在如图8所示的设备上运行。

图8为本发明一种基于卷积神经网络的车型识别设备的结构组成示意图。如图8所示，该设备800可包括：输入装置801、输出装置802、收发装置803、存储器804以及处理器805，其中：

所述输入装置801，用于接收外部访问控制设备的输入数据。具体实现中，本发明实施例所述的输入装置801可包括键盘、鼠标、光电输入装置、声音输入装置、触摸式输入装置、扫描仪等。

所述输出装置802，用于对外输出访问控制设备的输出数据。具体实现中，本发明实施例所述的输出装置802可包括显示器、扬声器、打印机等。

所述收发装置803，用于通过通信链路向其他设备发送数据或者从其他设备接收数据。具体实现中，本发明实施例的收发装置803可包括射频天线等收发器件。

所述存储器804，用于存储实现车型识别的程序。本发明实施例的存储器804可以是系统存储器，比如，非易失性的(诸如ROM，闪存等)，或者两者的结合。具体实现中，本发明实施例的存储器804还可以是系统之外的外部存储器，比如，磁盘、光盘、磁带等。

所述处理器805，用于运行所述存储器804中存储的实现车型识别的程序，以执行如下操作：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型。

进一步地，所述预设的图片数据包括预设的第一图片数据，所述对获取的待测图片进行预处理之前，包括：

将预设的第一图片数据分为第一训练集以及第一验证集；

利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型；

利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本；

若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型；

利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

进一步地，所述预设的图片数据还包括预设的第二图片数据，所述方法还包括：

将预设的第二图片数据分为第二训练集以及第二验证集；

利用所述第二训练集对第二卷积神经网络进行训练，以得到对应的第二中间模型；

利用所述第二验证集对所述第二中间模型进行验证，以得到对应的第二错误集，其中所述第二错误集包括至少一个第二错误样本；

若所述第二错误集中第二错误样本的数量大于或等于第二预设阀值，利用所述第二错误集训练所述第二中间模型以得到一个新的第二中间模型；

利用所述第二验证集对所述新的第二中间模型再次进行验证，直至所述第二错误集中第二错误样本的数量小于预设阀值，并判定此时的新的第二中间模型为对应的第二预设检测模型。

进一步地，所述第一卷积神经网络包括八层结构，所述第二卷积神经网络包括二十层结构，其中，所述第一卷积神经网络包括五个卷积层、两个全连接层以及一个用于二分类的概率统计层。

进一步地，所述将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息，包括：

将预处理后的待测图片输入第一预设检测模型以得到一置信度；

判断所述置信度是否大于预设临界值；

其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

本领域技术人员可以理解，图8中示出的基于卷积神经网络的车型识别设备的实施例并不构成对基于卷积神经网络的车型识别设备具体构成的限定，在其他实施例中，基于卷积神经网络的车型识别设备可以包括比图示更多或更少的部件，或者组合某些部件，或者不同的部件布置。例如，在一些实施例中，基于卷积神经网络的车型识别设备可以仅包括存储器及处理器，在这样的实施例中，存储器及处理器的结构及功能与图8所示实施例一致，在此不再赘述。

本发明提供了一种计算机可读存储介质，计算机可读存储介质存储有一个或者一个以上程序，所述一个或者一个以上程序可被一个或者一个以上的处理器执行，以实现以下步骤：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型。

进一步地，所述预设的图片数据包括预设的第一图片数据，所述对获取的待测图片进行预处理之前，包括：

将预设的第一图片数据分为第一训练集以及第一验证集；

利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型；

利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本；

若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型；

利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

进一步地，所述预设的图片数据还包括预设的第二图片数据，所述方法还包括：

将预设的第二图片数据分为第二训练集以及第二验证集；

利用所述第二训练集对第二卷积神经网络进行训练，以得到对应的第二中间模型；

利用所述第二验证集对所述第二中间模型进行验证，以得到对应的第二错误集，其中所述第二错误集包括至少一个第二错误样本；

若所述第二错误集中第二错误样本的数量大于或等于第二预设阀值，利用所述第二错误集训练所述第二中间模型以得到一个新的第二中间模型；

利用所述第二验证集对所述新的第二中间模型再次进行验证，直至所述第二错误集中第二错误样本的数量小于预设阀值，并判定此时的新的第二中间模型为对应的第二预设检测模型。

进一步地，所述第一卷积神经网络包括八层结构，所述第二卷积神经网络包括二十层结构，其中，所述第一卷积神经网络包括五个卷积层、两个全连接层以及一个用于二分类的概率统计层。

进一步地，所述将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息，包括：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以得到一置信度；

判断所述置信度是否大于预设临界值；

其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

本发明前述的存储介质包括：磁碟、光盘、只读存储记忆体(Read-Only Memory，ROM)等各种可以存储程序代码的介质。

本发明所有实施例中的单元可以通过通用集成电路，例如CPU(CentralProcessing Unit，中央处理器)，或通过ASIC(Application Specific IntegratedCircuit，专用集成电路)来实现。

本发明实施例方法中的步骤可以根据实际需要进行顺序调整、合并和删减。

本发明实施例装置中的单元可以根据实际需要进行合并、划分和删减。

所属领域的技术人员可以清楚地了解到，为了描述的方便和简洁，上述描述的设备、装置和单元的具体工作过程，可以参考前述方法实施例中的对应过程，在此不再赘述。

以上所述，仅为本发明的具体实施方式，但本发明的保护范围并不局限于此，任何熟悉本技术领域的技术人员在本发明揭露的技术范围内，可轻易想到各种等效的修改或替换，这些修改或替换都应涵盖在本发明的保护范围之内。因此，本发明的保护范围应以权利要求的保护范围为准。

Claims (10)

1.一种基于卷积神经网络的车型识别方法，其特征在于，所述方法包括：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。

2.如权利要求1所述的方法，其特征在于，所述预设的图片数据包括预设的第一图片数据，所述对获取的待测图片进行预处理之前，包括：

将预设的第一图片数据分为第一训练集以及第一验证集；

利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型；

利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本；

若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型；

利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

3.如权利要求2所述的方法，其特征在于，所述预设的图片数据还包括预设的第二图片数据，所述方法还包括：

将预设的第二图片数据分为第二训练集以及第二验证集；

利用所述第二训练集对第二卷积神经网络进行训练，以得到对应的第二中间模型；

利用所述第二验证集对所述第二中间模型进行验证，以得到对应的第二错误集，其中所述第二错误集包括至少一个第二错误样本；

若所述第二错误集中第二错误样本的数量大于或等于第二预设阀值，利用所述第二错误集训练所述第二中间模型以得到一个新的第二中间模型；

利用所述第二验证集对所述新的第二中间模型再次进行验证，直至所述第二错误集中第二错误样本的数量小于预设阀值，并判定此时的新的第二中间模型为对应的第二预设检测模型。

4.如权利要求3所述的方法，其特征在于，所述第一卷积神经网络包括八层结构，所述第二卷积神经网络包括二十层结构，其中，所述第一卷积神经网络包括五个卷积层、两个全连接层以及一个用于二分类的概率统计层。

5.如权利要求1所述的方法，其特征在于，所述将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息，包括：

将预处理后的待测图片输入第一预设检测模型以得到一置信度；

判断所述置信度是否大于预设临界值；

其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

6.一种基于卷积神经网络的车型识别装置，其特征在于，所述装置包括：

处理单元，用于对获取的待测图片进行预处理；

判断单元，用于将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

输入单元，用于若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

计算单元，用于通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定单元，用于确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。

7.如权利要求6所述的装置，其特征在于，所述预设的图片数据包括预设的第一图片数据，所述装置还包括：

分类单元，用于将预设的第一图片数据分为第一训练集以及第一验证集；

训练单元，用于利用所述第一训练集对第一卷积神经网络进行训练，以得到对应的第一中间模型；

验证单元，用于利用所述第一验证集对所述第一中间模型进行验证，以得到对应的第一错误集，其中所述第一错误集包括至少一个第一错误样本；

调整单元，用于若所述第一错误集中第一错误样本的数量大于或等于第一预设阀值，利用所述第一错误集训练所述第一中间模型以得到一个新的第一中间模型；

判定单元，用于利用所述第一验证集对所述新的第一中间模型再次进行验证，直至所述第一错误集中第一错误样本的数量小于预设阀值，并判定此时的新的第一中间模型为对应的第一预设检测模型。

8.如权利要求6所述的装置，其特征在于，所述判断单元包括：

置信度获取单元，用于将预处理后的待测图片输入第一预设检测模型以得到一置信度；

置信度判断单元，用于判断所述置信度是否大于预设临界值；

其中，若所述置信度大于预设临界值，所述待测图片含有车辆特征信息。

9.一种基于卷积神经网络的车型识别设备，其特征在于，包括：

存储器，用于存储实现车型识别的程序；以及

处理器，用于运行所述存储器中存储的实现车型识别的程序，以执行以下操作：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有一个或者一个以上程序，所述一个或者一个以上程序可被一个或者一个以上的处理器执行，以实现以下步骤：

对获取的待测图片进行预处理；

将预处理后的待测图片输入第一预设检测模型以判断所述待测图片是否含有车辆特征信息；

若所述待测图片含有车辆特征信息，将预处理后的待测图片输入第二预设检测模型；

通过所述第二预设检测模型计算得出所述待测图片对应于每一类车型的概率值；

确定所有概率值中的最大概率值，并将所述最大概率值对应的车型作为所述待测图片的车型；

其中，所述第一预设检测模型以及第二预设检测模型分别通过预设的图片数据对卷积神经网络进行相应的训练而获取。