[發明專利]輪胎信息獲取方法、裝置、計算機設備和存儲介質

在審

• 申請號: 202010320862.8
• 申請日: 2020-04-22
• 公開（公告）號: CN111507420A
• 公開（公告）日: 2020-08-07
• 發明（設計）人: 周康明;周佳敏
• 申請（專利權）人: 上海眼控科技股份有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/32
• 代理公司: 廣州華進聯合專利商標代理有限公司 44224
• 代理人: 李鵬

1.一種輪胎信息獲取方法，其特征在于，所述方法包括：

獲取待識別輪胎圖片中存在輪胎信息的待檢測子圖片；

采用預設的分類模型對所述待檢測子圖片進行顛倒類別分類，得到所述待檢測子圖片的目標顛倒類別；其中，所述分類模型為采用多個攜帶不同顛倒類別標簽的輪胎訓練圖片進行訓練所得到的模型；

若所述目標顛倒類別為顛倒，則將所述待檢測子圖片進行旋轉得到旋轉后的檢測子圖片；

采用預設的識別模型對所述旋轉后的檢測子圖片進行識別，得到所述輪胎信息。

2.根據權利要求1所述的方法，其特征在于，所述采用預設的分類模型對所述待檢測子圖片進行顛倒類別分類，得到所述待檢測子圖片的目標顛倒類別，包括：

獲取所述待檢測子圖片的顛倒置信度和不顛倒置信度；

若所述顛倒置信度大于所述不顛倒置信度，則確定所述待檢測子圖片的目標顛倒類別為顛倒；

若所述顛倒置信度小于所述不顛倒置信度，則確定所述待檢測子圖片的目標顛倒類別為不顛倒。

3.根據權利要求1或2所述的方法，其特征在于，所述分類模型的獲取過程包括：

獲取多個初始輪胎圖片；其中，多個所述初始輪胎圖片中包括針對輪胎在多種拍攝角度、多種光照強度所拍攝的圖片，且多個所述初始輪胎圖片為多種不同輪胎類型的圖片；

采用預設的目標檢測算法，對每個所述初始輪胎圖片進行檢測，得到每個初始輪胎圖片對應的存在輪胎信息的輪胎訓練圖片；其中，所述輪胎訓練圖片攜帶顛倒類別標簽；

將所述輪胎訓練圖片輸入預設的初始分類模型進行訓練，當所述初始分類模型輸出的損失函數滿足預設的收斂要求時，得到所述分類模型。

4.根據權利要求3所述的方法，其特征在于，所述將所述輪胎訓練圖片輸入預設的初始分類模型進行訓練之前，還包括：

當所述輪胎訓練圖片的數量小于預設的訓練圖片數量閾值時，將所述輪胎訓練圖片旋轉，得到旋轉訓練圖片；

將所述輪胎訓練圖片的顛倒類別標簽進行轉換，作為所述輪胎訓練圖片對應的旋轉訓練圖片的顛倒類別標簽；

將攜帶顛倒類別標簽的所述旋轉訓練圖片作為新增的輪胎訓練圖片。

5.根據權利要求1所述的方法，其特征在于，所述獲取待識別輪胎圖片中存在輪胎信息的待檢測子圖片，包括：

獲取待識別輪胎圖片；

采用預設的目標檢測算法對所述待識別輪胎圖片進行檢測，得到存在輪胎信息的所述待檢測子圖片。

6.根據權利要求1所述的方法，其特征在于，所述輪胎信息為表征輪胎規格的字符串，所述識別模型為文本識別模型；

所述采用預設的識別模型對所述旋轉后的檢測子圖片進行識別，得到所述輪胎信息，包括：

采用所述文本識別模型對所述旋轉后的檢測子圖片進行文本檢測，得到表征輪胎規格的目標字符串。

7.根據權利要求1所述的方法，其特征在于，所述方法還包括：

若所述目標顛倒類別為不顛倒，則采用所述識別模型對所述待檢測子圖片進行識別，得到所述輪胎信息。

8.一種輪胎信息獲取裝置，其特征在于，所述裝置包括：

獲取模塊，用于獲取待識別輪胎圖片中存在輪胎信息的待檢測子圖片；

分類模塊，用于采用預設的分類模型對所述待檢測子圖片進行顛倒類別分類，得到所述待檢測子圖片的目標顛倒類別；其中，所述分類模型為采用多個攜帶不同顛倒類別標簽的輪胎訓練圖片進行訓練所得到的模型；

處理模塊，用于當所述目標顛倒類別為顛倒時，則將所述待檢測子圖片進行旋轉得到旋轉后的檢測子圖片，并采用預設的識別模型對所述旋轉后的檢測子圖片進行識別，得到所述輪胎信息。

9.一種計算機設備，包括存儲器和處理器，所述存儲器存儲有計算機程序，其特征在于，所述處理器執行所述計算機程序時實現權利要求1至7中任一項所述的方法的步驟。

10.一種計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，所述計算機程序被處理器執行時實現權利要求1至7中任一項所述的方法的步驟。