[发明专利]车辆检测模型训练方法、设备、存储介质及装置

权利要求书

1.一种车辆检测模型训练方法，其特征在于，所述车辆检测模型训练方法包括以下步骤：

通过预设学生检测模型对车辆样本图像进行识别，获得第一特征；

通过预设教师检测模型对所述车辆样本图像进行识别，获得第二特征；

根据预设蒸馏损失函数计算所述第一特征与所述第二特征之间的损失参数；

根据所述损失参数更新所述预设学生检测模型中的待训练参数，以对所述学生检测模型进行训练，获得车辆检测模型。

2.如权利要求1所述的车辆检测模型训练方法，其特征在于，所述第一特征包括第一输出特征图和第一分布热力图，所述第二特征包括第二输出特征图和第二分布热力图，所述根据预设蒸馏损失函数计算所述第一输出特征与所述第二输出特征之间的损失参数，包括：

根据预设蒸馏损失函数计算所述第一输出特征图与所述第二输出特征图之间的特征图损失参数；

根据所述预设蒸馏损失函数计算所述第一分布热力图与所述第二分布热力图之间的热力图损失参数；

根据所述特征图损失参数以及所述热力图损失参数确定损失参数。

3.如权利要求2所述的车辆检测模型训练方法，其特征在于，所述根据所述特征图损失参数以及所述热力图损失参数确定损失参数，包括：

获取所述车辆样本图像的预设特征标记信息；

根据预设真实损失函数计算所述第二分布热力图与所述预设特征标记信息之间的真实损失参数；

根据所述特征图损失参数、所述热力图损失参数以及所述真实损失参数确定损失参数。

4.如权利要求2所述的车辆检测模型训练方法，其特征在于，分布热力图包括分类热力图、回归框宽高热力图以及中心偏移热力图；所述根据所述预设蒸馏损失函数计算所述第一分布热力图与所述第二分布热力图之间的热力图损失参数，包括：

根据所述预设蒸馏损失函数计算第一分类热力图与第二分类热力图之间的分类损失参数；

根据所述预设蒸馏损失函数计算第一回归框宽高热力图与第二回归框宽高热力图之间的回归框损失参数；

根据所述预设蒸馏损失函数计算第一中心偏移热力图与第二中心偏移热力图之间的中心偏移损失参数；

根据所述分类损失参数、所述回归框损失参数以及所述中心偏移损失参数确定热力图损失参数。

5.如权利要求1所述的车辆检测模型训练方法，其特征在于，所述通过预设学生检测模型对车辆样本图像进行识别，获得第一特征，包括：

通过预设学生检测模型对车辆样本图像进行识别，并获取所述预设学生检测模型中各输出层的输出特征；

在所述预设学生检测模型中各输出层的输出特征的特征数量大于预设特征数量阈值时，将所述预设学生检测模型中各输出层的输出特征作为第一特征。

6.如权利要求5所述的车辆检测模型训练方法，其特征在于，所述通过预设教师检测模型对所述车辆样本图像进行识别，获得第二特征，包括：

通过预设教师检测模型对车辆样本图像进行识别，并获取所述教师检测模型中各输出层的输出特征；

在所述教师检测模型中各输出层的输出特征的特征数量大于所述预设特征数量阈值时，将所述教师检测模型中各输出层的输出特征作为第二特征。

7.如权利要求1所述的车辆检测模型训练方法，其特征在于，所述根据所述损失参数更新所述预设学生检测模型中的待训练参数，以对所述学生检测模型进行训练，获得车辆检测模型之后，还包括：

获取待检测车辆图像，并通过所述车辆检测模型对所述待检测车辆图像进行检测，获得车辆行车参数；

将所述车辆行车参数与预设阈值进行比较，在所述车辆行车参数达到所述预设阈值时，进行报警。