[发明专利]一种确定车型的方法及相关装置

权利要求书

1.一种确定车型的方法，其特征在于，包括：

获取待处理车辆图像，所述待处理车辆图像包括车辆整体外观；

利用包括品牌分类子模型和车型分类子模型的目标车型分类模型，获得所述待处理车辆图像的品牌预测结果和车型预测结果；

基于所述品牌预测结果和所述车型预测结果，确定所述待处理车辆图像中车辆的车型。

2.根据权利要求1所述的方法，其特征在于，所述目标车型分类模型的获得步骤包括：

获取标注品牌标签、品牌+车型标签的训练车辆图像；

基于所述训练车辆图像和所述品牌标签训练第一分类网络，获得所述品牌分类子模型；基于所述训练车辆图像和所述品牌+车型标签训练第二分类网络，获得所述车型分类子模型；

基于所述品牌分类子模型和所述车型分类子模型，获得所述目标车型分类模型。

3.根据权利要求1所述的方法，其特征在于，所述利用包括品牌分类子模型和车型分类子模型的目标车型分类模型，获得所述待处理车辆图像的品牌预测结果和车型预测结果，包括：

将待处理车辆图像输入所述车辆检测模型获得车辆区域图像；

将所述车辆区域图像输入所述品牌分类子模型获得所述品牌预测结果；将所述车辆区域图像输入所述车型分类子模型获得所述车型预测结果。

4.根据权利要求1-3任一项所述的方法，其特征在于，所述品牌预测结果包括预测品牌和对应的预测品牌得分；所述车型预测结果包括预测车型和对应的预测车型得分。

5.根据权利要求4所述的方法，其特征在于，所述基于所述品牌预测结果和所述车型预测结果，确定所述待处理车辆图像中车辆的车型，包括：

基于所述预测品牌、所述预测品牌得分、所述预测车型和所述预测车型得分，获得所述预测车型对应的目标预测车型得分；

基于所述预测车型对应的目标预测车型得分，确定所述待处理车辆图像中车辆的车型。

6.根据权利要求5所述的方法，其特征在于，所述基于所述预测品牌、所述预测品牌得分、所述预测车型和所述预测车型得分，获得所述预测车型对应的目标预测车型得分，具体为：

按照所述预测品牌和所述预测车型的对应关系，将所述预测品牌得分和对应的所述预测车型得分相乘，获得所述预测车型对应的目标预测车型得分。

7.根据权利要求5所述的方法，其特征在于，所述基于所述预测品牌、所述预测品牌得分、所述预测车型和所述预测车型得分，获得所述预测车型对应的目标预测车型得分，具体为：

基于所述预测品牌、所述预测品牌得分、所述预测车型和所述预测车型得分，输入融合神经网络获得所述预测车型对应的目标预测车型得分。

8.一种确定车型的装置，其特征在于，包括：

获取单元，用于获取待处理车辆图像，所述待处理车辆图像包括车辆整体外观；

获得单元，用于利用包括品牌分类子模型和车型分类子模型的目标车型分类模型，获得所述待处理车辆图像的品牌预测结果和车型预测结果；

确定单元，用于基于所述品牌预测结果和所述车型预测结果，确定所述待处理车辆图像中车辆的车型。

9.一种用于确定车型的装置，其特征在于，包括有存储器，以及一个或者一个以上的程序，其中一个或者一个以上程序存储于存储器中，且经配置以由一个或者一个以上处理器执行所述一个或者一个以上程序包含用于进行以下操作的指令：

获取待处理车辆图像，所述待处理车辆图像包括车辆整体外观；

利用包括品牌分类子模型和车型分类子模型的目标车型分类模型，获得所述待处理车辆图像的品牌预测结果和车型预测结果；

基于所述品牌预测结果和所述车型预测结果，确定所述待处理车辆图像中车辆的车型。

10.一种机器可读介质，其上存储有指令，当由一个或多个处理器执行时，使得装置执行如权利要求1至7中任一项所述的确定车型的方法。