[發明專利]一種基于二維標簽進行汽車標注與識別的AI算法

在審

• 申請號: 201811634845.0
• 申請日: 2018-12-29
• 公開（公告）號: CN109766808A
• 公開（公告）日: 2019-05-17
• 發明（設計）人: 顧巖;肖光磊;葉朱宇;黃嘉偉
• 申請（專利權）人: 南通貝法瑞信息科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62
• 代理公司: 暫無信息
• 代理人: 暫無信息

1.一種基于二維標簽進行汽車標注與識別的AI算法，其特征在于：包括以下步驟：

步驟S1、構建二維汽車標簽，是對收集的汽車圖片進行標注，二維汽車標簽包括第一維的汽車品牌標簽和第二維的汽車零件標簽；

步驟S2、基于二維標簽的汽車品牌標注，通過兩個維度的標簽，假設第一維標簽集為N1，第二維標簽集為N2，則汽車品牌的標簽集達到N1 x N2；

步驟S3、在收集的汽車圖片上標注出汽車的局部特征區域，對于每個標注數據還需要附帶汽車的品牌、型號信息，形成jpg和png圖片格式的汽車識別訓練集；

步驟S4、在訓練數據目錄下存放作為模型的訓練圖片數據，并和自有的預標注數據結合進行訓練；

步驟S5、基于二維標簽的汽車品牌標注訓練后生成的模型，在利用模型識別影視中的圖片的汽車品牌時，識別的原始結果包括了一維標簽中汽車品牌以及相似度，還包括了二維標簽中的汽車零件相似度值，再根據兩個維度的相似度，根據權重于擬合算法，重新計算出與汽車品牌的相似度，從而判斷出視頻每幀中出現的汽車品牌、汽車區域。

2.根據權利要求1所述的一種基于二維標簽進行汽車標注與識別的AI算法，其特征在于：訓練、識別時采用卷積神經網絡對輸入的圖片進行特征提取，識別時會根據提取的特征輸出識別的分類以及相似度。

3.根據權利要求1所述的一種基于二維標簽進行汽車標注與識別的AI算法，其特征在于：步驟S4中，訓練前需要將各個數據集的數據與標注格式統一為訓練框架要求的格式，通過腳本可以將各個數據集的格式轉換為框架能識別的數據格式。

4.根據權利要求1所述的一種基于二維標簽進行汽車標注與識別的AI算法，其特征在于：汽車零件的標簽為車燈、車頭、車尾、車門、車頂或車標。