一種車輛屬性識別模型的訓練方法、系統(tǒng)及相關設備

基本信息

申請?zhí)?/td> CN202110827957.3 申請日 -
公開（公告）號 CN113283549A 公開（公告）日 2021-08-20
申請公布號 CN113283549A 申請公布日 2021-08-20
分類號 G06K9/62（2006.01）I;G08G1/017（2006.01）I;G06N3/04（2006.01）I 分類 計算；推算；計數;
發(fā)明人 馬永軍;周東暉;閆瀟寧 申請（專利權）人 深圳市安軟慧視科技有限公司
代理機構 深圳君信誠知識產權代理事務所（普通合伙） 代理人 劉偉
地址 518000廣東省深圳市龍華新區(qū)龍華辦事處清祥路清湖工業(yè)園寶能科技園7棟A座6樓ACDF
法律狀態(tài) -

摘要

摘要 本發(fā)明提供了一種車輛屬性識別模型的訓練方法、系統(tǒng)及相關設備，訓練方法包括以下步驟：構建包括可縮放的主干網絡、BiFPN網絡、跨類別注意力網絡和屬性分類網絡的車輛屬性識別模型；獲取用于模型訓練的圖片集，其中圖片包含車輛屬性，并對圖片進行數據增強，得到增強圖片集；將增強圖片集輸入到主干網絡中，得到初級特征圖集；將初級特征圖集利用BiFPN網絡進行融合，得到融合特征圖；將融合特征圖輸入到跨類別注意力網絡，對其中圖片的車輛屬性進行權值標記和屬性間信息提取，得到第三特征圖集；將第三特征圖集輸入到屬性分類網絡，得到車輛屬性識別結果。本發(fā)明使用跨類別的注意力機制構建車輛屬性識別模型，提升了模型的泛化能力。