一种车牌定位识别模型的训练方法及相关装置与流程

本申请涉及图像处理领域，具体涉及一种车牌定位识别模型的训练方法及相关装置。

背景技术：

1、随着科技的飞速发展，汽车领域逐步在从低级别的辅助驾驶过渡到高级别的自动驾驶，而为了能够让自动驾驶应对复杂多变的路况和行车场景，需要对车辆图像数据进行相应的收集和分析处理，以便来对车辆进行改进研发、优化用户体验、个性化服务等目的，此时，由于车辆图像数据中经常会包括有个人信息的车牌，故为了数据安全考虑，需要对车辆图像数据进行车牌脱敏之后才能进行相应的分析处理，其中，如何准确对车辆图像数据进行车牌定位识别对车牌脱敏来说至关重要。

2、相关技术中主要是通过对车牌字符进行分割识别来对车辆图像数据实现车牌定位识别，具体来说，对车辆图像数据进行预处理之后，可以先对车牌进行初步定位，然后再对车牌字符进行分割，通过对分割后的车牌字符进行识别来实现对车牌的识别。

3、虽然对车牌字符进行分割识别能够实现对车牌的准确定位识别，但是对车牌进行分割识别的步骤较为复杂，进而导致定位识别效率较低。鉴于此，如何训练得到一种能够快速定位识别车牌的车牌定位识别模型是亟需解决的问题。

技术实现思路

1、本申请实施例提供了一种车牌定位识别模型的训练方法及相关装置，能够训练得到一种能够快速定位识别车牌的车牌定位识别模型。

2、有鉴于此，本申请实施例第一方面提供一种车牌定位识别模型的训练方法，包括：

3、获取车牌样本图像和对应的真实车牌位置数据，车牌样本图像为包括车牌的样本图像，真实车牌位置数据用于标识对应的车牌样本图像包括的车牌的真实位置；

4、将车牌样本图像输入初始车牌定位识别模型，得到对应的预测车牌位置数据；

5、根据预测车牌位置数据、真实车牌位置数据和预先设定的损失函数，调整初始车牌定位识别模型的模型参数，得到车牌定位识别模型，车牌定位识别模型用于对包括车牌的车辆图片进行车牌定位识别处理。

6、本申请实施例第二方面提供一种车牌定位识别模型的训练装置，包括：

7、获取单元，用于获取车牌样本图像和对应的真实车牌位置数据，车牌样本图像为包括车牌的样本图像，真实车牌位置数据用于标识对应的车牌样本图像包括的车牌的真实位置；

8、确定单元，用于将车牌样本图像输入初始车牌定位识别模型，得到对应的预测车牌位置数据；

9、调整单元，用于根据预测车牌位置数据、真实车牌位置数据和预先设定的损失函数，调整初始车牌定位识别模型的模型参数，得到车牌定位识别模型，车牌定位识别模型用于对包括车牌的车辆图片进行车牌定位识别处理。

10、本申请实施例第三方面提供一种电子设备，包括：

11、存储器，用于存储可执行指令；

12、处理器，用于执行所述存储器中存储的可执行指令时，实现本申请实施例提供的车牌定位识别模型的训练方法。

13、本申请实施例第四方面提供一种计算机可读介质，存储有可执行指令，用于被处理器执行时，实现本申请实施例提供的车牌定位识别模型的训练方法。

14、本申请实施例提供了一种车牌定位识别模型的训练方法及相关装置，该方法包括：获取车牌样本图像和对应的真实车牌位置数据，车牌样本图像为包括车牌的样本图像，真实车牌位置数据用于标识对应的车牌样本图像包括的车牌的真实位置；将车牌样本图像输入初始车牌定位识别模型，得到对应的预测车牌位置数据；根据预测车牌位置数据、真实车牌位置数据和预先设定的损失函数，调整初始车牌定位识别模型的模型参数，得到车牌定位识别模型，车牌定位识别模型用于对包括车牌的车辆图片进行车牌定位识别处理。通过上述方法，可以训练得到一种能够快速准确定位识别图像中车牌的车牌定位识别模型。

技术特征：

1.一种车牌定位识别模型的训练方法，其特征在于，包括：

2.根据权利要求1所述的方法，其特征在于，所述真实车牌位置数据包括真实车牌矩形框的中心点坐标、真实车牌矩形框的宽度和真实车牌矩形框的高度；

3.根据权利要求2所述的方法，其特征在于，所述损失函数包括矩形框中心坐标损失函数、矩形框宽高损失函数和置信度损失函数；

4.根据权利要求1所述的方法，其特征在于，在将所述车牌样本图像输入初始车牌定位识别模型之前，所述方法还包括对所述车牌样本图像进行数据增强处理。

5.根据权利要求1所述的方法，其特征在于，所述方法还包括：

6.根据权利要求5所述的方法，其特征在于，当所述车牌定位识别模型包括多个不同尺度的目标检测层时，所述根据所述目标待脱敏车牌图像，通过所述车牌定位识别模型进行车牌定位识别处理，得到目标预测车牌位置数据，包括：

7.根据权利要求5所述的方法，其特征在于，所述基于所述目标预测车牌位置数据，对所述目标待脱敏车牌图像中的待脱敏车牌进行脱敏处理，包括：

8.一种车牌定位识别模型的训练装置，其特征在于，包括：

9.一种电子设备，其特征在于，包括：

10.一种计算机可读存储介质，其特征在于，存储有可执行指令，用于被处理器执行时，实现权利要求1至7任一项所述的车牌定位识别模型的训练方法。

技术总结本申请提供了一种车牌定位识别模型的训练方法及相关装置，该方法包括：获取车牌样本图像和对应的真实车牌位置数据，车牌样本图像为包括车牌的样本图像，真实车牌位置数据用于标识对应的车牌样本图像包括的车牌的真实位置；将车牌样本图像输入初始车牌定位识别模型，得到对应的预测车牌位置数据；根据预测车牌位置数据、真实车牌位置数据和预先设定的损失函数，调整初始车牌定位识别模型的模型参数，得到车牌定位识别模型，车牌定位识别模型用于对包括车牌的车辆图片进行车牌定位识别处理。通过上述方法，可以训练得到一种能够快速准确定位识别图像中车牌的车牌定位识别模型。技术研发人员：曹玮鑫,田忠鹏,张晓瑞受保护的技术使用者：上海汽车集团股份有限公司技术研发日：技术公布日：2024/9/2