[發明專利]車輛檢測及標注方法、系統及電子設備

在審

• 申請號: 202110361601.5
• 申請日: 2021-04-02
• 公開（公告）號: CN113033449A
• 公開（公告）日: 2021-06-25
• 發明（設計）人: 楊旭波;戴方越;李霖;安康;徐瑋怡
• 申請（專利權）人: 上海國際汽車城（集團）有限公司;上海交通大學
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/46;G08G1/017;G06N3/04
• 代理公司: 上海光華專利事務所(普通合伙) 31219
• 代理人: 龐紅芳

本發明提供一種車輛檢測及標注方法、系統及電子設備，所述車輛檢測及標注系統包括：視頻流追蹤系統和視頻流標注系統；所述視頻流追蹤系統用于追蹤一段視頻流中出現的車輛軌跡，逐幀得到視頻流中車輛的位置坐標；所述視頻流標注系統用于對追蹤出的車輛進行車型及顏色的標記，并標記視頻中同一輛車出現的所有幀。本發明可以充分利用車輛識別和追蹤技術的優勢，在保證可用的標注準確率的同時，免去逐幀標記視頻流中車輛的繁瑣和效率低下。

技術領域

本發明涉及車輛識別、車輛追蹤及數據標注領域，特別設計一種車輛檢測及標注方法、 系統及電子設備。

背景技術

隨著近年來圖像識別和深度學習技術的不斷發展，許多領域需要訓練自己的汽車識別模 型。但由于各種領域對汽車識別模型的需求不同，很多情況下需要對已有的模型進行數據集 增強，或者重新用車輛數據訓練自己的模型。

傳統的車輛數據標注采用的是用圖像或者視頻逐幀進行手工標注。在例如LabelImg、 OpenCV/CVAT、Ybat等工具的標注過程中，標注者需要逐幀在圖像上勾畫車輛的包圍盒或 輪廓。這樣的標注優勢是準確率高，但是也存在一個十分明顯的缺點，那就是效率較為低下。

如果使用車輛追蹤技術對追蹤出的車輛進行標注，在建立軌跡聯系后只要對一個識別框 進行標注，后續所有的連續的識別框都能被標注上。理想情況下一個視頻中出現了幾輛車， 就只需要進行幾次標注。一輛車常常在一個視頻中出現數百幀甚至更多，使用車輛追蹤技術 無疑能大幅提高標注效率。

發明內容

鑒于以上所述現有技術的缺點，本發明的目的在于提供一種車輛檢測及標注方法、系統 及電子設備，用于解決現有技術無法車輛數據標注效率低、費時費力的技術問題。

為實現上述目的及其他相關目的，本發明提供一種車輛檢測及標注方法，包括：追蹤一 段視頻流中出現的車輛軌跡，逐幀得到視頻流中車輛的位置坐標；在視頻流被追蹤完畢后， 創建一個可被視頻流標注系統讀取的項目，并為后續的標注提供車輛的追蹤信息；對追蹤出 的車輛進行車型及顏色的標記，并標記視頻中同一輛車出現的所有幀。

于本發明的一實施例中，所述追蹤一段視頻流中出現的車輛軌跡，逐幀得到視頻流中車 輛的位置坐標包括：根據車輛識別模型輸出的位置坐標，以及視頻流中的車輛特征信息進行 匹配及預測；逐幀識別視頻流中車輛的位置坐標。

于本發明的一實施例中，所述標記視頻中同一輛車出現的所有幀包括：標注追蹤出的一 條車輛軌跡；手工標注未追蹤出的車輛軌跡；刪除錯誤的車輛追蹤包圍盒、切割軌跡連續的 數輛車。

于本發明的一實施例中，所述車輛檢測及標注方法還包括：將視頻剪切成原視頻的一部 分；根據識別出的幀序號和車輛位置坐標，對視頻進行渲染，并使用亮色標記出視頻中的車 輛信息。

本發明的實施例還提供一種車輛檢測及標注系統，包括：視頻流追蹤系統和視頻流標注 系統；所述視頻流追蹤系統用于追蹤一段視頻流中出現的車輛軌跡，逐幀得到視頻流中車輛 的位置坐標；所述視頻流標注系統用于對追蹤出的車輛進行車型及顏色的標記，并標記視頻 中同一輛車出現的所有幀。

于本發明的一實施例中，還包括分別與所述視頻流追蹤系統和視頻流標注系統連接的車 輛數據保存系統；所述車輛數據保存系統用于在視頻流被追蹤完畢后，創建一個可被所述視 頻流標注系統讀取的項目，并為后續的標注提供車輛的追蹤信息。

于本發明的一實施例中，所述視頻流追蹤系統包括車輛識別模型和車輛追蹤模型；所述 車輛識別模型用于逐幀識別視頻流中車輛的位置坐標；所述車輛追蹤模型用于根據車輛識別 模型輸出的位置坐標，以及視頻流中的車輛特征信息進行匹配及預測。