[發明專利]識別模型訓練方法、識別方法、裝置、設備及存儲介質

在審

• 申請號: 202110591890.8
• 申請日: 2021-05-28
• 公開（公告）號: CN113326773A
• 公開（公告）日: 2021-08-31
• 發明（設計）人: 蘇翔博;王健;孫昊
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/04
• 代理公司: 北京市鑄成律師事務所 11313
• 代理人: 郭麗祥;曹遠

本公開提供了識別模型訓練方法、識別方法、裝置、設備及存儲介質，深度學習、計算機視覺技術領域。具體實現方案為：將待處理圖像輸入待訓練的識別模型；通過所述待訓練的識別模型的特征輸出層，輸出所述待處理圖像的至少一張特征圖；通過所述待訓練的識別模型的頭部，根據所述至少一張特征圖獲取所述待處理圖像中的第一目標物體的預測數據，以及所述待處理圖像中的與所述第一目標物體的預測數據關聯的第二目標物體的預測數據；根據第一目標物體的預測數據、第二目標物體的預測數據、第一目標物體的標注數據和第二目標物體的標注數據，優化所述待訓練的識別模型，得到訓練后的識別模型。本公開實施能夠提高識別模型的識別效率和識別效果。

技術領域

本公開涉及人工智能技術領域，尤其涉及深度學習、計算機視覺技術領域，可應用于智慧城市、智能交通場景下。

背景技術

目標識別是圖像處理的一個重要的手段和目的，通過目標識別，可以對視頻、靜態畫面中的物體、人體、動物體等目標物體進行識別，根據識別結果實現身份認證、安全檢查等多種用途。

隨著計算機技術的發展，在多種需要應用到目標識別技術的場景下，隨著應用目的的多樣化，被識別的目標物體的多樣化，經常需要多重模型實現目標物體識別的目的。如何提高處理待處理圖像的模型的效率，是需要改進的一個問題。

發明內容

本公開提供了一種識別模型訓練方法、識別方法、裝置、設備及存儲介質。

根據本公開的一方面，提供了一種識別模型訓練方法，包括：

將待處理圖像輸入待訓練的識別模型；

通過待訓練的識別模型的特征輸出層，輸出待處理圖像的至少一張特征圖；

通過待訓練的識別模型的頭部，根據至少一張特征圖獲取待處理圖像中的第一目標物體的預測數據，以及待處理圖像中的與第一目標物體的預測數據關聯的第二目標物體的預測數據；

根據第一目標物體的預測數據、第二目標物體的預測數據、第一目標物體的標注數據和第二目標物體的標注數據，優化待訓練的識別模型，得到訓練后的識別模型。

根據本公開的另一方面，提供了一種識別方法，包括：

將待識別圖像輸入識別模型，獲得待識別圖像中的第一目標物體的預測數據和與第一目標物體關聯的第二目標物體的預測數據，識別模型為本公開任意一項實施例所提供的訓練后的識別模型。

根據本公開的另一方面，提供了一種識別模型訓練裝置，包括：

第一輸入模塊，用于將待處理圖像輸入待訓練的識別模型；

特征圖模塊，用于通過待訓練的識別模型的特征輸出層，輸出待處理圖像的至少一張特征圖；

預測數據模塊，用于通過待訓練的識別模型的頭部，根據至少一張特征圖獲取待處理圖像中的第一目標物體的預測數據，以及待處理圖像中的與第一目標物體的預測數據關聯的第二目標物體的預測數據；

訓練模塊，用于根據第一目標物體的預測數據、第二目標物體的預測數據、第一目標物體的標注數據和第二目標物體的標注數據，優化待訓練的識別模型，得到訓練后的識別模型。

根據本公開的另一方面，提供了一種識別裝置，包括：

第二輸入模塊，用于將待識別圖像輸入識別模型，獲得待識別圖像中的第一目標物體的預測數據和與第一目標物體關聯的第二目標物體的預測數據，識別模型為本公開任意一項實施例所提供的訓練后的識別模型。

根據本公開的另一方面，提供了一種電子設備，包括：

至少一個處理器；以及

與該至少一個處理器通信連接的存儲器；其中，