[發明專利]車輛重識別方法、電子設備及相關產品

在審

• 申請號: 202110063645.X
• 申請日: 2021-01-18
• 公開（公告）號: CN112766281A
• 公開（公告）日: 2021-05-07
• 發明（設計）人: 范艷;張鵬;賀武;吳偉華
• 申請（專利權）人: 深圳市華尊科技股份有限公司
• 主分類號: G06K9/46
• 分類號: G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 熊永強

本申請實施例公開了一種車輛重識別方法、電子設備及相關產品，應用于電子設備，該方法包括：分別將第一查詢圖像和第一底庫圖像輸入到預設風格轉換模型，得到第二查詢圖像和第二底庫圖像；將第一車輛訓練數據輸入到預設風格轉換模型，得到第二車輛訓練數據；通過第一車輛訓練數據和第二車輛訓練數據訓練第一特征提取網絡，得到第二特征提取網絡；將第二查詢圖像輸入到第二特征提取網絡進行特征提取，得到查詢圖像特征；將第二底庫圖像輸入到第二特征提取網絡進行特征提取，得到底庫圖像特征；將查詢圖像特征和底庫圖像特征進行匹配，得到匹配結果，展示匹配結果。采用本申請實施例可以提升車輛重識別準確率。

技術領域

本申請涉及圖像處理技術領域，具體涉及一種車輛重識別方法、電子設備及相關產品。

背景技術

車輛重識別是指給定一張車輛圖像，識別出車輛圖像數據庫中的同一車輛，在真實交通監控系統中，車輛重識別可以起到對目標車輛進行定位、監管、刑偵的作用。車輛重識別任務中通常會有多個攝像頭，并且由于各種因素(環境，光線等)，同一輛車的每個攝像頭拍攝的圖像風格通常會有所不同。此外，由于時間(早晨，中午，下午等)的不同，即使是同一臺攝像機也會拍攝不同風格的圖像。因此，圖像的風格會發生變化，這對最終車輛重識別任務的結果有很大影響。另外，攝像頭與車輛之間相對位置的變化，會導致同一車輛在不同圖像中呈現不同的角度，車輛的外觀存在著顯著的變化，此外，由于具有不同身份的車輛可以具有相同的品牌，型號和顏色，提取具有判別力的特征對車輛重識別的任務非常的關鍵，進而，如何提升真實交通監控場景中車輛重識別的準確率問題亟待解決。

發明內容

本申請實施例提供了一種車輛重識別方法及相關產品，能夠提升車輛重識別準確率。

第一方面，本申請實施例提供一種車輛重識別方法，應用于電子設備，所述方法包括：

獲取第一查詢圖像、第一底庫圖像和第一車輛訓練數據；

分別將所述第一查詢圖像和所述第一底庫圖像輸入到預設風格轉換模型，得到第二查詢圖像和第二底庫圖像，所述預設風格轉換模型用于將輸入數據轉化為風格統一的輸出數據；

將所述第一車輛訓練數據輸入到所述預設風格轉換模型，得到第二車輛訓練數據；

通過所述第一車輛訓練數據和所述第二車輛訓練數據訓練第一特征提取網絡，得到第二特征提取網絡；

將所述第二查詢圖像輸入到所述第二特征提取網絡進行特征提取，得到查詢圖像特征；

將所述第二底庫圖像輸入到所述第二特征提取網絡進行特征提取，得到底庫圖像特征；

將所述查詢圖像特征和所述底庫圖像特征進行匹配，得到匹配結果，展示所述匹配結果。

第二方面，本申請實施例提供一種車輛重識別裝置，應用于電子設備，所述裝置包括：獲取單元、轉換單元、訓練單元、提取單元和展示單元，其中，

所述獲取單元，用于獲取第一查詢圖像、第一底庫圖像和第一車輛訓練數據；

所述轉換單元，用于分別將所述第一查詢圖像和所述第一底庫圖像輸入到預設風格轉換模型，得到第二查詢圖像和第二底庫圖像，所述預設風格轉換模型用于將輸入數據轉化為風格統一的輸出數據；

所述轉換單元，還用于將所述第一車輛訓練數據輸入到所述預設風格轉換模型，得到第二車輛訓練數據；

所述訓練單元，用于通過所述第一車輛訓練數據和所述第二車輛訓練數據訓練第一特征提取網絡，得到第二特征提取網絡；

所述提取單元，用于將所述第二查詢圖像輸入到所述第二特征提取網絡進行特征提取，得到查詢圖像特征；

所述提取單元，還用于將所述第二底庫圖像輸入到所述第二特征提取網絡進行特征提取，得到底庫圖像特征；