[发明专利]一种车辆颜色识别方法、装置、设备及存储介质

权利要求书

1.一种车辆颜色识别方法，其特征在于，包括：

根据接收到的待识别车辆图像，提取第一车身图像；

根据所述第一车身图像，通过预先建立的车辆颜色识别模型，获取所述第一车身图像对应的初始识别颜色；

当所述初始识别颜色不属于预设的反光色时，将所述初始识别颜色输出为最终识别颜色；

当所述初始识别颜色属于所述反光色时，将所述第一车身图像转换为HSV图像；

根据所述HSV图像和所述初始识别颜色对应的HSV参数范围，获得所述第一车身图像对应的二值掩码图；所述HSV参数范围是指在HSV颜色空间中能表示所述初始识别颜色所属的类型的参数的上下阈值之间的参数范围；

判断所述二值掩码图中像素值为1的像素点总数是否大于预设阈值；

当所述二值掩码图中像素值为1的像素点总数大于预设阈值，将所述初始识别颜色输出为最终识别颜色；

当所述二值掩码图中像素值为1的像素点总数不大于预设阈值，根据所述二值掩码图，通过所述车辆颜色识别模型，获取所述第一车身图像对应的最终识别颜色。

2.如权利要求1所述的车辆颜色识别方法，其特征在于，所述根据接收到的待识别车辆图像，提取第一车身图像，具体包括：

对所述待识别车辆图像进行裁剪，获得第一车身图像。

3.如权利要求1所述的车辆颜色识别方法，其特征在于，所述方法还包括：

将预先采集的车辆图片按照预先设定的颜色矩阵进行颜色分类，获得若干个训练集；其中，所述颜色矩阵包括多种颜色及各种颜色所属的颜色类型；

通过所述训练集，对预先构建的深度神经网络模型进行训练，得到所述车辆颜色识别模型；其中，预先构建的所述深度神经网络模型用于提取浅层特征。

4.如权利要求1所述的车辆颜色识别方法，其特征在于，所述根据所述HSV图像和所述初始识别颜色对应的HSV参数范围，获得所述第一车身图像对应的二值掩码图，具体包括：

判断所述HSV图像中的像素点的HSV参数是否在所述初始识别颜色对应的HSV参数范围内；HSV参数指的是HSV图像对应的HSV颜色空间模型中的色调(H)、饱和度(S)、明度(V)；

当判断出所述HSV图像中像素点的HSV参数在所述初始识别颜色对应的HSV参数范围内时，将所述HSV图像中像素点的像素值更新为1；

当判断出所述HSV图像中像素点的HSV参数不在所述初始识别颜色对应的HSV参数范围内时，将所述HSV图像中像素点的像素值更新为0；

根据像素值更新后的HSV图像，得到所述二值掩码图。

5.如权利要求1所述的车辆颜色识别方法，其特征在于，所述当所述二值掩码图中像素值为1的像素点总数不大于预设阈值，根据所述二值掩码图，通过所述车辆颜色识别模型，获取所述第一车身图像对应的最终识别颜色，具体包括：

根据所述二值掩码图中像素值为1的像素点，剔除所述第一车身图像中与像素值为1的像素点对应像素点颜色信息，获得第二车身图像；

根据所述第二车身图像，通过所述车辆颜色识别模型，获取所述第一车身图像对应的最终识别颜色。