[發明專利]一種車型分類方法及裝置

在審

• 申請號: 201911102143.2
• 申請日: 2019-11-12
• 公開（公告）號: CN112861567A
• 公開（公告）日: 2021-05-28
• 發明（設計）人: 扈霽
• 申請（專利權）人: 上海高德威智能交通系統有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/20;G06K9/32;G06K9/34;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京柏杉松知識產權代理事務所(普通合伙) 11413
• 代理人: 馬敬;李欣

本申請提供了一種車型分類方法及裝置，屬于計算機技術領域，所述應用于電子設備，所述方法包括：通過預設的粗粒度分類模型，對包含待識別機動車輛的圖片進行初步分類，得到所述待識別機動車輛的第一車型；如果所述第一車型為預設車型，則通過預設的細粒度分類模型，對所述包含待識別機動車輛的圖片進行二次分類，得到所述待識別機動車輛的第二車型，其中，所述細粒度分類模型為基于圖像樣本及其所包括的機動車輛的車型預先訓練完成的，所述細粒度分類模型包括圖像特征與機動車輛的車型的對應關系；根據所述第二車型，確定所述待識別機動車輛的目標車型。采用本申請，能夠提高車型分類的準確率。

技術領域

本申請涉及計算機技術領域，特別是涉及一種車型分類方法及裝置。

背景技術

交通管理部門針對機動車輛的不同車型，設置有不同的管理規定，管理規定例如禁止行駛的路線、允許停放的位置。其中，車型包括轎車、面包車、客車、公交車以及皮卡車。為了保證道路安全，可以通過識別機動車輛的車型的方式，對違反管理規定的機車車輛進行檢測。

相關技術中，用于檢測違反管理規定的機動車輛的電子設備，可以通過安裝在道路兩側的監控攝像機獲取包含待識別機動車輛的圖片。然后，電子設備可以通過預設的卷積神經網絡和該圖片，對待識別機動車輛的車型進行分類，得到待識別機動車輛的目標車型。之后，電子設備可以判斷目標車型是否為預設車型，如果目標車型是預設車型，則電子設備可以判定待識別機動車輛違反了管理規定。

然而，通過卷積神經網絡和包含待識別機動車輛的圖片，無法對外形相似的客車和公交車進行分類，因此，基于作為粗粒度分類模型的卷積神經網絡對機動車輛的車型進行分類，導致車型分類的準確率低。

發明內容

本申請實施例的目的在于提供一種車型分類方法及裝置，以實現提高車型分類的準確率。具體技術方案如下：

第一方面，提供了一種車型分類方法，應用于電子設備，所述方法包括：

通過預設的粗粒度分類模型，對包含待識別機動車輛的圖片進行初步分類，得到所述待識別機動車輛的第一車型；

如果所述第一車型為預設車型，則通過預設的細粒度分類模型，對所述包含待識別機動車輛的圖片進行二次分類，得到所述待識別機動車輛的第二車型，其中，所述細粒度分類模型為基于圖像樣本及其所包括的機動車輛的車型預先訓練完成的，所述細粒度分類模型包括機動車輛各部位的圖像特征與機動車輛的車型的對應關系；

根據所述第二車型，確定所述待識別機動車輛的目標車型。

可選的，如果所述第一車型為預設車型，所述方法還包括：

從所述包含待識別機動車輛的圖片中，提取所述待識別機動車輛的車臉圖像區域；

識別所述車臉圖像區域中包含的字符信息；

根據預先存儲的字符信息與車型的對應關系、所述字符信息，確定所述待識別機動車輛的第三車型；

所述根據所述第二車型，確定所述待識別機動車輛的目標車型包括：

如果所述第二車型與所述第三車型相同，將所述第二車型作為所述待識別機動車輛的目標車型。

可選的，如果所述第二車型與所述第三車型不同，所述方法還包括：

基于所述細粒度分類模型和所述包含待識別機動車輛的圖像，確定所述第二車型的第一置信水平；

基于所述字符信息與車型的對應關系、所述字符信息，確定所述第三車型的第二置信水平；

如果所述第一置信水平大于所述第二置信水平，則將所述第二車型作為所述待識別機動車輛的目標車型；

如果所述第一置信水平小于所述第二置信水平，則將所述第三車型作為所述待識別機動車輛的目標車型。