[发明专利]轮胎信息获取方法、装置、计算机设备和存储介质

说明书

本申请涉及一种轮胎信息获取方法、装置、计算机设备和存储介质。所述方法包括：获取待识别轮胎图片中存在轮胎信息的待检测子图片；采用预设的分类模型对所述待检测子图片进行颠倒类别分类，得到所述待检测子图片的目标颠倒类别；其中，所述分类模型为采用多个携带不同颠倒类别标签的轮胎训练图片进行训练所得到的模型；若所述目标颠倒类别为颠倒，则将所述待检测子图片进行旋转得到旋转后的检测子图片；采用预设的识别模型对所述旋转后的检测子图片进行识别，得到所述轮胎信息。采用本方法能够提高轮胎信息的获取效率。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种轮胎信息获取方法、装置、计算机设备和存储介质。

背景技术

随着科学技术的发展，人们日常的工作和生活已经逐渐的实现了自动化。例如在车辆年审的时候，常常采用检测设备对车辆的轮胎的规格进行自动检查，来实现车辆的轮胎检测。

通常车辆年审过程中，是通过拍摄车辆的轮胎的图像，然后采用检测设备识别轮胎的图像中印制有轮胎规格的文本，从而得到轮胎的规格。这一过程中，由于不能保证每次车辆停止的位置刚好能够使得轮胎规格为正向，所以导致图片不能保证每次都被正常识别。传统的做法是，当图像中的文本识别失败的时候，则将图像进行旋转后再次重新识别。

然而，传统的将识别失败的图像旋转后重新识别的方式，导致部分图片需要识别两次，因此识别效率低。

发明内容

基于此，有必要针对上述技术问题，提供一种高效的轮胎信息获取方法、装置、计算机设备和存储介质。

一种轮胎信息获取方法，所述方法包括：

获取待识别轮胎图片中存在轮胎信息的待检测子图片；

采用预设的分类模型对所述待检测子图片进行颠倒类别分类，得到所述待检测子图片的目标颠倒类别；其中，所述分类模型为采用多个携带不同颠倒类别标签的轮胎训练图片进行训练所得到的模型；

若所述目标颠倒类别为颠倒，则将所述待检测子图片进行旋转得到旋转后的检测子图片；

采用预设的识别模型对所述旋转后的检测子图片进行识别，得到所述轮胎信息。

在其中一个实施例中，所述采用预设的分类模型对所述待检测子图片进行颠倒类别分类，得到所述待检测子图片的目标颠倒类别，包括：

获取所述待检测子图片的颠倒置信度和不颠倒置信度；

若所述颠倒置信度大于所述不颠倒置信度，则确定所述待检测子图片的目标颠倒类别为颠倒；

若所述颠倒置信度小于所述不颠倒置信度，则确定所述待检测子图片的目标颠倒类别为不颠倒。

在其中一个实施例中，所述分类模型的获取过程包括：

获取多个初始轮胎图片；其中，多个所述初始轮胎图片中包括针对轮胎在多种拍摄角度、多种光照强度所拍摄的图片，且多个所述初始轮胎图片为多种不同轮胎类型的图片；

采用预设的目标检测算法，对每个所述初始轮胎图片进行检测，得到每个初始轮胎图片对应的存在轮胎信息的轮胎训练图片；其中，所述轮胎训练图片携带颠倒类别标签；

将所述轮胎训练图片输入预设的初始分类模型进行训练，当所述初始分类模型输出的损失函数满足预设的收敛要求时，得到所述分类模型。

在其中一个实施例中，所述将所述轮胎训练图片输入预设的初始分类模型进行训练之前，还包括：

当所述轮胎训练图片的数量小于预设的训练图片数量阈值时，将所述轮胎训练图片旋转，得到旋转训练图片；

将所述轮胎训练图片的颠倒类别标签进行转换，作为所述轮胎训练图片对应的旋转训练图片的颠倒类别标签；