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

在審

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

本發明公開了一種基于二維標簽進行汽車標注與識別的AI算法，構建二維汽車標簽，二維標簽包括第一維汽車品牌標簽和第二維汽車零件標簽；構建汽車品牌的標簽集；基于二維標簽的汽車品牌標注訓練后生成的模型，在利用模型識別影視中的圖片的汽車品牌時，識別的原始結果包括了一維標簽中汽車品牌以及相似度，還包括了二維標簽中的汽車零件相似度值，再根據兩個維度的相似度，根據權重于擬合算法，重新計算出與汽車品牌的相似度，從而判斷出視頻每幀中出現的汽車品牌、汽車區域。有效解決了常規AI方法無法有效識別影視中遠鏡頭和有遮擋難以識別汽車品牌的問題。

技術領域

本發明涉及一種AI算法，具體涉及一種基于二維標簽進行汽車標注與識別的AI算法。

背景技術

汽車品牌識別，越來越普遍的應用于很多方面，譬如二手車估價等，快速準備識別汽車品牌輔助于汽車相關行業的智能化應用，將AI技術應用于汽車品牌的識別也是風起云涌。

現有的汽車品牌AI識別，是通過圖片標注的汽車品牌標簽進行標注，汽車品牌的標簽都是一維標簽，如奧迪RS-3、奧迪TT RS、奧迪 Q5、奧迪Q7、奧迪A8、寶馬E90、寶馬F10、保時捷Cayenne、保時捷Macan、保時捷911等。

標注汽車品牌，通過一維汽車品牌的標簽，對圖片中出現汽車的整體進行標注，標注的標簽是上述種的各種汽車品牌。在對所有的汽車圖片標注完成后，有AI框架和選擇合適的網絡進行訓練，生成汽車品牌模型用于圖片中的汽車識別。

通過一維汽車品牌的標簽，進行標注和AI訓練生成汽車識別的模型，生成的AI模型可以識別汽車品牌，如奧迪RS-3、奧迪TT RS、奧迪 Q5、奧迪Q7、奧迪A8、寶馬E90、寶馬F10、保時捷Cayenne、保時捷Macan、保時捷911等。通過一維標簽生成的AI模型能識別比較清晰的圖片，如手機對著汽車拍出的完整汽車的圖片，對這類清晰而且汽車完整畫面的圖片識別準確率較高。

但是對于影視中的汽車品牌識別準確率很低，因為影視中出現的汽車，大多數都是遠鏡頭，畫面中的汽車會顯得很小；即使是近鏡頭時，畫面往往也會被人物遮擋，不是完整的汽車畫面。

現有汽車品牌識別的缺點：

（1）分類特征不明顯：在分類識別問題中，不同分類之間特征差距越大越好。對汽車整體進行標注，沒有指定區分各品牌的關鍵特征，會導致不同品牌汽車的相似特征也會被提取出來，不利于進行分類，從而降低識別的準確率。

（2）識別角度魯棒性差：圖片中的整車在不同角度下特征區別較大。標注時如果只選一個角度的圖片，會導致識別時只在這個角度下準確率較高而其他角度時不準。標注時如果選擇多個角度的圖片，則由于不同角度的特征區別較大，會導致這一類汽車的特征混亂（也就是無特征）。

（3）對待識別圖片中汽車的完整性要求較高：由于標注以及訓練時是汽車的整體特征，如果待識別圖片中的汽車如果有部分遮擋，會使提取的特征缺失從而導致識別不準確。

發明內容

為解決上述背景技術中的問題，本發明提供一種基于二維標簽進行汽車標注與識別的AI算法。

本發明提供如下技術方案：

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

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

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

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