[發明專利]識別模型的訓練方法、裝置、計算機設備及識別方法

在審

• 申請號: 201910880226.8
• 申請日: 2019-09-18
• 公開（公告）號: CN110619330A
• 公開（公告）日: 2019-12-27
• 發明（設計）人: 周康明;谷維鑫
• 申請（專利權）人: 上海眼控科技股份有限公司
• 主分類號: G06K9/32
• 分類號: G06K9/32;G06K9/34;G06K9/62
• 代理公司: 44224 廣州華進聯合專利商標代理有限公司
• 代理人: 黃麗霞

1.一種車輛識別碼的識別模型的訓練方法，其特征在于，所述方法包括：

獲取多個車輛識別碼區域圖像，并標注各個所述車輛識別碼區域圖像中各個字符的位置和類別；

對標注后的各個所述車輛識別碼區域圖像進行數據擴充操作，生成各個所述車輛識別碼區域圖像所對應的擴充圖像，并確定各個所述擴充圖像中各個字符的位置和類別；

根據各個所述車輛識別碼區域圖像以及各個所述擴充圖像，生成車輛識別碼區域圖像的樣本集；

根據所述車輛識別碼區域圖像的樣本集、各個所述車輛識別碼區域圖像中各個字符的位置和類別、各個所述擴充圖像中各個字符的位置和類別，訓練所述車輛識別碼識別模型。

2.根據權利要求1所述的方法，其特征在于，所述數據擴充操作包括布局變換；所述對標注后的各個所述車輛識別碼區域圖像進行數據擴充操作，生成各個所述車輛識別碼區域圖像所對應的擴充圖像，包括：

對各個所述車輛識別碼區域圖像分別進行截取，獲得不包含所述各個字符的背景圖像；

調整所述背景圖像的尺寸；

根據各個字符在各個所述車輛識別碼區域圖像中的位置，在預設位置處對各個所述車輛識別碼區域圖像中的車輛識別碼進行截斷，獲得至少兩部分的車輛識別碼圖像；

至少兩部分的車輛識別碼圖像疊加至尺寸調整后的背景圖像，生成布局變換操作后的擴充圖像。

3.根據權利要求2所述的方法，其特征在于，所述將至少兩部分的車輛識別碼圖像疊加至尺寸調整后的背景圖像，生成布局變換操作后的擴充圖像，包括：

分別調整至少兩部分的車輛識別碼圖像的尺寸；

將尺寸調整后的至少兩部分的車輛識別碼圖像疊加至尺寸調整后的背景圖像，生成布局變換操作后的擴充圖像。

4.根據權利要求1所述的方法，其特征在于，所述數據擴充操作包括彎曲變換；所述對標注后的各個所述車輛識別碼區域圖像進行數據擴充操作，生成各個所述車輛識別碼區域圖像所對應的擴充圖像，包括：

根據各個所述車輛識別碼區域圖像中第一個字符與最后一個字符的位置，在各個所述車輛識別碼區域圖像中分別確定圍繞車輛識別碼的外接矩形框；

獲取各個所述外接矩形框各頂點的坐標和各個所述車輛識別碼區域圖像的中心坐標；

根據弧形車輛識別碼的實際弧度確定彎曲變換所對應的彎曲系數；

根據所述各頂點的坐標和所述中心坐標，結合所述彎曲系數對各個所述車輛識別碼區域圖像進行彎曲變換，生成彎曲變換操作后的擴充圖像。

5.根據權利要求1所述的方法，其特征在于，所述數據擴充操作包括旋轉、平移中的至少一個；所述對標注后的各個所述車輛識別碼區域圖像進行數據擴充操作，生成各個所述車輛識別碼區域圖像所對應的擴充圖像，包括：

對標注后的各個所述車輛識別碼區域圖像進行預設角度的旋轉，生成旋轉操作后的擴充圖像；和/或

按照預設方向對標注后的各個所述車輛識別碼區域圖像進行平移，生成平移操作后的擴充圖像。

6.根據權利要求1所述的方法，其特征在于，所述車輛識別碼識別模型采用字符實例分割模型。

7.根據權利要求6所述的方法，其特征在于，利用所述字符實例分割模型識別待識別車輛識別碼區域圖像中的車輛識別碼，包括：

獲取所述待識別車輛識別碼區域圖像和所述車輛識別碼的參考字符串；

通過所述字符實例分割模型對所述待識別車輛識別碼區域圖像進行識別分割；

根據所述待識別車輛識別碼區域圖像的識別分割結果生成所述待識別車輛識別碼區域圖像所對應的字符串；

將所述待識別車輛識別碼區域圖像所對應的字符串與所述車輛識別碼的參考字符串進行比對；

根據比對的結果，記錄所述車輛識別碼的識別結果。

8.根據權利要求1所述的方法，其特征在于，在所述對標注后的各個所述車輛識別碼區域圖像進行數據擴充操作，生成各個所述車輛識別碼區域圖像所對應的擴充圖像之后，所述方法還包括：

檢測所述擴充圖像是否包含殘缺字符；

若包含，則刪除包含殘缺字符的擴充圖像。