[發明專利]車輛識別方法、裝置、計算機設備和存儲介質

在審

• 申請號: 202010766113.8
• 申請日: 2020-08-03
• 公開（公告）號: CN112001269A
• 公開（公告）日: 2020-11-27
• 發明（設計）人: 肖鐘雯
• 申請（專利權）人: 浙江大華技術股份有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/46;G06K9/62
• 代理公司: 杭州華進聯浙知識產權代理有限公司 33250
• 代理人: 陳涵

1.一種車輛識別方法，其特征在于，包括：

獲取車輛圖片；

將所述車輛圖片輸入特種車輛識別模型，得到特種車輛預判結果；

將所述車輛圖片輸入標志物識別模型，得到標志物識別結果；

根據所述特種車輛預判結果和標志物識別結果，輸出最終車輛識別結果。

2.根據權利要求1所述的車輛識別方法，其特征在于，所述將所述車輛圖片輸入特種車輛識別模型，獲取特種車輛預判結果包括：

將所述車輛圖片輸入特種車輛識別模型，得到各特種車輛類別的第一置信度；

根據各特種車輛類別的第一置信度，獲取最大第一置信度對應的特征車輛類別；

將最大第一置信度對應的特種車輛類別以及對應的第一置信度作為所述特種車輛預判結果。

3.根據權利要求2所述的車輛識別方法，其特征在于，所述將所述車輛圖片輸入標志物識別模型，獲取標志物識別結果包括：

將所述車輛圖片輸入標志物識別模型，得到至少一種標志物以及所述標志物對應的第二置信度。

4.根據權利要求3所述的車輛識別方法，其特征在于，所述根據所述預判結果和標志物識別結果，輸出最終車輛識別結果包括：

若所述特種車輛預判結果中的第一置信度大于等于第一閾值時，則將特種車輛預判結果中的特種車輛類別作為最終車輛識別結果。

5.根據權利要求3所述的車輛識別方法，其特征在于，所述根據所述特種車輛預判結果和標志物識別結果，輸出最終車輛識別結果還包括：

若所述特種車輛預判結果中的第一置信度小于第一閾值時，獲取標志物識別結果中的至少一種標志物以及所述標志物對應的第二置信度；

若第二置信度大于等于第二閾值，則獲取第二置信度對應的標志物，并且將所述標志物對應的車輛類別作為最終車輛識別結果。

6.根據權利要求5所述的車輛識別方法，其特征在于，所述若第二置信度大于等于第二閾值，則獲取第二置信度對應的標志物，并且將所述標志物對應的車輛類別作為最終車輛識別結果包括：

若第二置信度小于第二閾值，則將特種車輛預判結果中的特種車輛類別作為最終車輛識別結果。

7.根據權利要求1所述的車輛識別方法，其特征在于，特種車輛識別模型以及標志物識別模型均為通過對卷積神經網絡訓練得到。

8.一種車輛識別裝置，其特征在于，所述裝置包括：

車輛獲取模塊，得到所述車輛圖片；

特種車輛識別模塊，將所述車輛圖片輸入特種車輛識別模型，得到特種車輛預判結果；

標志物識別模塊，將所述車輛圖片輸入標志物識別模型，得到標志物識別結果；

特種車輛類別決策模塊，根據所述特種車輛預判結果和標志物識別結果，輸出最終車輛識別結果。

9.一種計算機設備，包括存儲器、處理器以及存儲在所述存儲器上并可在所述處理器上運行的計算機程序，其特征在于，所述處理器執行所述計算機程序時實現如權利要求1至7中任一項所述的車輛識別方法。

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