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

说明书

说明书披露一种车牌检测模型的训练方法和装置，方法包括：利用以下步骤进行迭代训练，直至满足所述车牌检测模型的模型训练要求：将样本图像输入所述车牌检测模型，得到所述样本图像的车牌预测框，所述样本图像标注有车牌标注框；根据所述车牌预测框和所述车牌标注框确定角度维度下的第一旋转损失；根据所述车牌预测框和所述车牌标注框确定面积维度下的第二旋转损失；根据所述第一旋转损失和所述第二旋转损失确定综合旋转损失；根据所述综合旋转损失更新所述车牌检测模型的模型参数。

技术领域

本说明书涉及人工智能领域，尤其涉及一种车牌检测模型的训练方法和装置。

背景技术

车牌是车辆的重要标识，在公路、停车场等的监控系统中，都可以借助人工智能来对车牌进行识别，以根据车牌号确定车辆和车主信息。车牌识别前往往需要进行车牌检测，车牌检测是从复杂的图像中检测到车牌所在的位置。基于检测到的车牌位置进而进行车牌识别。车牌检测是车牌识别的基础，如何提高车牌检测的准确率成了亟待解决的问题。

发明内容

有鉴于此，本说明书提供一种车牌检测模型的训练方法和装置。

具体地，本说明书是通过如下技术方案实现的：

根据本申请的第一方面，提供一种车牌检测模型的训练方法，所述方法包括：

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

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

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

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

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

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

根据本申请的第二方面，提供一种车牌检测模型的训练装置，所述装置包括：

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

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

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

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

综合确定单元，用于根据所述第一旋转损失和所述第二旋转损失确定综合旋转损失；

第一参数更新单元，用于根据所述综合旋转损失更新所述车牌检测模型的模型参数。

根据本申请的第三方面，提供一种车牌检测模型的训练装置，所述装置包括：

处理器；

用于存储机器可执行指令的存储器；

其中，通过读取并执行所述存储器存储的与车牌检测模型的训练逻辑对应的机器可执行指令，所述处理器被促使：

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

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

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

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