[發明專利]優化車牌檢測模型的方法及系統、車牌檢測方法及系統

有效

• 申請號: 202010134676.5
• 申請日: 2020-03-02
• 公開（公告）號: CN111401360B
• 公開（公告）日: 2023-06-20
• 發明（設計）人: 張浩
• 申請（專利權）人: 杭州雄邁集成電路技術股份有限公司
• 主分類號: G06V20/62
• 分類號: G06V20/62;G06V30/148;G06V30/19;G06V10/764;G06N3/08;G06N3/0464;G06N3/048
• 代理公司: 杭州裕陽聯合專利代理有限公司 33289
• 代理人: 姚宇吉

1.一種優化車牌檢測模型的方法，其特征在于包括以下步驟：

獲取待優化車牌檢測網絡，并獲取用于訓練所述待優化車牌檢測網絡的樣本圖像及其第一標注信息；

在所述待優化車牌檢測網絡中添加車牌頭尾字符標識分支網絡，生成待訓練網絡；

在所述第一標注信息中添加與樣本圖像相對應的車牌頭部字符框和車牌尾部字符框，生成第二標注信息；

利用樣本圖像及其第二標注信息對所述待訓練網絡進行迭代訓練，獲得優化網絡，將所述優化網絡中車牌頭尾字符標識分支網絡進行移除，獲得優化后的車牌檢測模型；

其中，利用樣本圖像及其第二標注信息對所述待訓練網絡進行迭代訓練，獲得優化網絡的具體步驟為：

獲取所述待訓練網絡輸出的預測數據，所述預測數據包括和樣本圖像相對應的熱點圖和字符標識預測數據；

根據所述第二標注信息和所述預測數據計算預測損失數據，所述預測損失數據包括字符標識預測損失值；

基于所述預測損失數據對所述待訓練網絡進行反饋更新，直至完成對所述待訓練網絡的訓練，獲得優化網絡；

其中，計算所述字符標識預測損失值的步驟為：

提取所述熱點圖和所述字符標識預測數據；

根據對應樣本圖像的車牌頭部字符框、車牌尾部字符框、熱點圖和字符標識預測數據計算所述車牌頭尾字符標識分支網絡的損失值，獲得字符標識預測損失值，具體為：

將所述車牌頭部字符框和所述車牌尾部字符框映射至所述熱點圖中，獲得映射字符框；

從所述映射字符框中提取若干點作為字符正樣本，并按照預設的正負困難樣本比例從所述熱點圖中挖掘對應數量的字符負樣本；

為所述字符正樣本和所述字符負樣本標注字符標識真實數據，并提取所述字符正樣本和所述字符負樣本標注字符的字符標識預測數據，根據所述字符標識真實數據和所述字符標識預測數據進行損失值計算，獲得字符標識預測損失值。

2.根據權利要求1所述的優化車牌檢測模型的方法，其特征在于：

第一標注信息包括樣本圖像中各車牌的車牌框和類別；

預測損失數據還包括分類預測損失值和位置預測損失值。

3.一種優化車牌檢測模型的系統，其特征在于，包括：

獲取模塊，用于獲取待優化車牌檢測網絡，還用于獲取用于訓練所述待優化車牌檢測網絡的樣本圖像及其第一標注信息；

分支網絡添加模塊，用于在所述待優化車牌檢測網絡中添加車牌頭尾字符標識分支網絡，生成待訓練網絡；

標注添加模塊，用于在所述第一標注信息中添加與樣本圖像相對應的車牌頭部字符框和車牌尾部字符框，生成第二標注信息；

優化模塊，用于利用樣本圖像及其第二標注信息對所述待訓練網絡進行迭代訓練，獲得優化網絡，將所述優化網絡中車牌頭尾字符標識分支網絡進行移除，獲得優化后的車牌檢測模型；

所述優化模塊包括預測數據獲取單元、計算單元、反饋更新單元和移除單元：

所述預測數據獲取單元，用于獲取所述待訓練網絡輸出的預測數據，所述預測數據包括和樣本圖像相對應的熱點圖和字符標識預測數據；

所述計算單元，用于根據所述第二標注信息和所述預測數據計算預測損失數據，所述預測損失數據包括字符標識預測損失值；

所述反饋更新單元，用于基于所述預測損失數據對所述待訓練網絡進行反饋更新，直至完成對所述待訓練網絡的訓練，獲得優化網絡；

所述計算單元被配置為：

提取所述熱點圖和所述字符標識預測數據；

根據對應樣本圖像的車牌頭部字符框、車牌尾部字符框、熱點圖和字符標識預測數據計算所述車牌頭尾字符標識分支網絡的損失值，獲得字符標識預測損失值，具體為：

將所述車牌頭部字符框和所述車牌尾部字符框映射至所述熱點圖中，獲得映射字符框；

從所述映射字符框中提取若干點作為字符正樣本，并按照預設的正負困難樣本比例從所述熱點圖中挖掘對應數量的字符負樣本；

為所述字符正樣本和所述字符負樣本標注字符標識真實數據，并提取所述字符正樣本和所述字符負樣本標注字符的字符標識預測數據，根據所述字符標識真實數據和所述字符標識預測數據進行損失值計算，獲得字符標識預測損失值。

4.一種計算機可讀存儲介質，其存儲有計算機程序，其特征在于，該程序被處理器執行時實現權利要求1或2所述方法的步驟。