[发明专利]车辆重识别方法、电子设备及相关产品

说明书

本申请实施例公开了一种车辆重识别方法、电子设备及相关产品，应用于电子设备，该方法包括：分别将第一查询图像和第一底库图像输入到预设风格转换模型，得到第二查询图像和第二底库图像；将第一车辆训练数据输入到预设风格转换模型，得到第二车辆训练数据；通过第一车辆训练数据和第二车辆训练数据训练第一特征提取网络，得到第二特征提取网络；将第二查询图像输入到第二特征提取网络进行特征提取，得到查询图像特征；将第二底库图像输入到第二特征提取网络进行特征提取，得到底库图像特征；将查询图像特征和底库图像特征进行匹配，得到匹配结果，展示匹配结果。采用本申请实施例可以提升车辆重识别准确率。

技术领域

本申请涉及图像处理技术领域，具体涉及一种车辆重识别方法、电子设备及相关产品。

背景技术

车辆重识别是指给定一张车辆图像，识别出车辆图像数据库中的同一车辆，在真实交通监控系统中，车辆重识别可以起到对目标车辆进行定位、监管、刑侦的作用。车辆重识别任务中通常会有多个摄像头，并且由于各种因素(环境，光线等)，同一辆车的每个摄像头拍摄的图像风格通常会有所不同。此外，由于时间(早晨，中午，下午等)的不同，即使是同一台摄像机也会拍摄不同风格的图像。因此，图像的风格会发生变化，这对最终车辆重识别任务的结果有很大影响。另外，摄像头与车辆之间相对位置的变化，会导致同一车辆在不同图像中呈现不同的角度，车辆的外观存在着显著的变化，此外，由于具有不同身份的车辆可以具有相同的品牌，型号和颜色，提取具有判别力的特征对车辆重识别的任务非常的关键，进而，如何提升真实交通监控场景中车辆重识别的准确率问题亟待解决。

发明内容

本申请实施例提供了一种车辆重识别方法及相关产品，能够提升车辆重识别准确率。

第一方面，本申请实施例提供一种车辆重识别方法，应用于电子设备，所述方法包括：

获取第一查询图像、第一底库图像和第一车辆训练数据；

分别将所述第一查询图像和所述第一底库图像输入到预设风格转换模型，得到第二查询图像和第二底库图像，所述预设风格转换模型用于将输入数据转化为风格统一的输出数据；

将所述第一车辆训练数据输入到所述预设风格转换模型，得到第二车辆训练数据；

通过所述第一车辆训练数据和所述第二车辆训练数据训练第一特征提取网络，得到第二特征提取网络；

将所述第二查询图像输入到所述第二特征提取网络进行特征提取，得到查询图像特征；

将所述第二底库图像输入到所述第二特征提取网络进行特征提取，得到底库图像特征；

将所述查询图像特征和所述底库图像特征进行匹配，得到匹配结果，展示所述匹配结果。

第二方面，本申请实施例提供一种车辆重识别装置，应用于电子设备，所述装置包括：获取单元、转换单元、训练单元、提取单元和展示单元，其中，

所述获取单元，用于获取第一查询图像、第一底库图像和第一车辆训练数据；

所述转换单元，用于分别将所述第一查询图像和所述第一底库图像输入到预设风格转换模型，得到第二查询图像和第二底库图像，所述预设风格转换模型用于将输入数据转化为风格统一的输出数据；

所述转换单元，还用于将所述第一车辆训练数据输入到所述预设风格转换模型，得到第二车辆训练数据；

所述训练单元，用于通过所述第一车辆训练数据和所述第二车辆训练数据训练第一特征提取网络，得到第二特征提取网络；

所述提取单元，用于将所述第二查询图像输入到所述第二特征提取网络进行特征提取，得到查询图像特征；

所述提取单元，还用于将所述第二底库图像输入到所述第二特征提取网络进行特征提取，得到底库图像特征；