[发明专利]一种融合MobileNet-V2和svm的车辆重识别方法

权利要求书

1.一种融合MobileNet-V2和svm的车辆重识别方法，其特征在于，基于MobileNet-V2的深度学习网络，搭建多分支网络并进行多分支网络训练，在训练过程中基于svm方法得到一个车辆重识别模型，基于所述车辆重识别模型对目标车辆做出检索结果展示。

2.根据权利要求1所述的融合MobileNet-V2和svm的车辆重识别方法，其特征在于，所述重识别方法的具体步骤包括如下：

步骤一为训练数据预处理：将样本数据进行图像预处理，之后再对经过图像预处理的样本数据进行数据标定；

步骤二为多分支网络构建：以MobileNet-V2为基础网络搭建多分支网络，所述多分支网络包括全局特征分支网络以及车辆部件分支网络；

步骤三为多分支网络训练：首先，将训练图像和样本标签送入搭建好的车辆部件分支网络进行训练，当损失函数达到收敛，得到车辆部件检测模型，然后添加全局特征分支网络以调整车辆部件检测模型，当损失函数达到收敛，得到车辆重识别模型，即完成一次计算训练过程，模型训练时的N个计算训练过程对应得到N个车辆重识别模型，之后基于N个车辆重识别模型进行svm权重筛选得到最终的车辆重识别模型；

步骤四为识别效果展示：基于最终的车辆重识别模型，搭建调用代码，调用代码包括读入一张图像，然后进行图像预处理，然后加载所述最终的车辆重识别模型进行特征提取，把所有图像经过重复上述操作，可以得到每个图像的特征，存入文件A中，再搭建匹配代码，匹配代码包括读入一张图像，然后进行图像预处理，然后加载所述最终的车辆重识别模型进行特征提取，导入文件A中，进行特征相似度计算，取相似度高的前100张图，把得到的100张图写到一个大图中进行结果展示。

3.根据权利要求2所述的融合MobileNet-V2和svm的车辆重识别方法，其特征在于，所述步骤一中，图像预处理包括有图像缩放、图像格式转换、图像旋转；

图像缩放是采用双线性插值算法对获得的抓拍图像进行缩放，模拟得到不同实际场景下的抓拍图；

图像格式转换是基于转换公式将不同格式的图像转换呈统一的格式；

图像旋转是基于笛卡尔坐标系以图像的重心为圆心进行旋转。

4.根据权利要求3所述的融合MobileNet-V2和svm的车辆重识别方法，其特征在于，所述数据标定设置为区域框标定，可包括车脸区域框标定、车前灯区域框标定、车轮框标定、车辆后视镜标定。

5.根据权利要求4所述的融合MobileNet-V2和svm的车辆重识别方法，其特征在于，各个区域框标定时，框要完整包括整个对应的车辆零件，当该车辆零件显现的区域少于整个车轮面积的三分之一时，不做标定。

6.根据权利要求2所述的融合MobileNet-V2和svm的车辆重识别方法，其特征在于，在所述步骤三中，得到的车辆部件检测模型在测试集上进行效果验证，测试集标定了不同场景下的车辆各个子区域位置框作为答案，把车辆部件检测模型预测出来的检测框和子区域位置框进行匹配，根据不同的重合度阈值的准确率来判断当前的车辆部件检测模型是否达到预期效果。

7.根据权利要求6所述的融合MobileNet-V2和svm的车辆重识别方法，其特征在于，在所述步骤三中，得到的车辆重识别模型在测试集上进行效果验证，测试集标定了M个ID车辆，每个ID下面有多张不同场景下相机拍摄的同一辆车，然后还有N个随机车辆图像。对N个随机车辆图像以及每个ID对应的所有车辆图像，用得到的车辆重识别模型提取特征，存放到一个文件A中，然后对标定的M个ID下的一张车辆图像进行逐一提取特征，把提取到的特征和文件A进行比对，根据相似度得分由高到低的顺序进行排序，取前100个，最后查看每个ID下的图否搜到其他剩余的车辆图像，且排序靠前，统计Top1、Top5、Top10的准确率，以此来验证车辆重识别模型的效果。

8.根据权利要求7所述的融合MobileNet-V2和svm的车辆重识别方法，其特征在于，在所述步骤三中，svm权重筛选是对N个车辆重识别模型的每一层特征进行筛选，得到最好的每层特征权重，然后把得到的各层最终的特征权重重新写入车辆重识别模型中，即可得到最终的车辆重识别最终模型。