[发明专利]车牌检测模型的训练方法和装置

权利要求书

1.一种车牌检测模型的训练方法，其特征在于，所述方法包括：

利用以下步骤进行迭代训练，直至满足所述车牌检测模型的模型训练要求：

将样本图像输入所述车牌检测模型，得到所述样本图像的车牌预测框，所述样本图像标注有车牌标注框；

根据所述车牌预测框和所述车牌标注框确定角度维度下的第一旋转损失；

根据所述车牌预测框和所述车牌标注框确定面积维度下的第二旋转损失；

根据所述第一旋转损失和所述第二旋转损失确定综合旋转损失；

根据所述综合旋转损失更新所述车牌检测模型的模型参数；

所述根据所述车牌预测框和所述车牌标注框确定面积维度下的第二旋转损失，包括：

确定所述车牌预测框的最小外接矩形，作为车牌预测包围框；

确定所述车牌标注框的最小外接矩形，作为车牌标注包围框；

根据所述车牌预测包围框和所述车牌标注包围框确定所述第二旋转损失；

所述车牌检测模型包括：特征提取层、特征筛选层和车牌检测层；

所述特征提取层，用于获取所述样本图像，并对所述样本图像进行特征提取，得到基础特征图；

所述特征筛选层，用于对所述基础特征图进行特征筛选，得到热点特征图；

所述车牌检测层，用于根据所述基础特征图和所述热点特征图预测得到所述车牌预测框；

所述方法还包括：

根据所述车牌标注框和所述热点特征图确定特征筛选损失；

根据所述特征筛选损失更新所述特征筛选层的模型参数；

根据所述综合旋转损失更新车牌检测层的模型参数；

根据所述特征筛选损失和所述综合旋转损失更新所述特征提取层的模型参数。

2.根据权利要求1所述的方法，其特征在于，所述根据所述车牌预测框和所述车牌标注框确定角度维度下的第一旋转损失，包括：

确定基准点与所述车牌预测框的相对顶点之间的第一角度；

确定所述基准点与所述车牌标注框的所述相对顶点之间的第二角度；

根据所述第一角度和所述第二角度确定所述第一旋转损失。

3.根据权利要求2所述的方法，其特征在于，所述根据所述第一角度和所述第二角度确定所述第一旋转损失，包括：

确定位置关系相同的所述第一角度和所述第二角度之间的角度差值；

根据位置关系不同的角度差值确定所述第一旋转损失。

4.根据权利要求2所述的方法，其特征在于，所述基准点位于所述车牌预测框对角线所在的直线上；或

所述基准点位于所述车牌标注框对角线所在的直线上。