[發(fā)明專利]目標檢測模型的訓練方法、目標檢測方法及其相關設備

有效

• 申請?zhí)枺?/td>: 202111279491.4
• 申請日: 2021-10-29
• 公開（公告）號: CN114092874B
• 公開（公告）日: 2023-07-25
• 發(fā)明（設計）人: 張偉;譚嘯;孫昊
• 申請（專利權）人: 北京百度網(wǎng)訊科技有限公司
• 主分類號: G06V20/52
• 分類號: G06V20/52;G06V10/26;G06V10/774
• 代理公司: 北京清亦華知識產(chǎn)權代理事務所(普通合伙) 11201
• 代理人: 單冠飛

本公開提供了目標檢測模型的訓練方法、目標檢測方法及其相關設備，涉及人工智能技術領域，具體涉及計算機視覺和深度學習技術領域。方案為：采用標注有第一部件分割標簽的分割訓練圖像對部件分割模型進行訓練，并采用標注有第一深度值的深度訓練圖像對深度估計模型進行訓練；通過針對每一幀深度訓練圖像采用經(jīng)過訓練的部件分割模型標注第二部件分割標簽，并針對每一幀分割訓練圖像采用經(jīng)過訓練的深度估計模型標注第二深度值，生成目標檢測模型訓練集；采用目標檢測模型訓練集，對用于深度預測和部件分割的目標檢測模型進行訓練。由此，可使目標檢測模型實現(xiàn)深度預測與部件分割，并且在目標檢測模型的訓練過程中可以兼容單一來源的標注樣本。

技術領域

本公開涉及人工智能技術領域，具體涉及計算機視覺和深度學習技術領域，尤其涉及目標檢測模型的訓練方法、目標檢測方法及其相關設備。

背景技術

在智慧城市、智能交通、視頻分析場景下，對視頻或者圖像中的車輛、行人、物體等事物或目標進行精準檢測，可以為車輛統(tǒng)計等任務提供幫助。因此，如何實現(xiàn)視頻或者圖像中的目標檢測是非常重要的。

發(fā)明內(nèi)容

本公開提供了一種用于目標檢測模型的訓練方法、目標檢測方法及其相關設備。

根據(jù)本公開的一方面，提供了一種目標檢測模型的訓練方法，包括：獲取標注有第一部件分割標簽的分割訓練圖像，和標注有第一深度值的深度訓練圖像；采用所述標注有第一部件分割標簽的分割訓練圖像對部件分割模型進行訓練，以及采用所述標注有第一深度值的深度訓練圖像對深度估計模型進行訓練；針對每一幀所述深度訓練圖像采用經(jīng)過訓練的部件分割模型標注第二部件分割標簽，并針對每一幀分割訓練圖像采用經(jīng)過訓練的深度估計模型標注第二深度值；根據(jù)具有所述第二部件分割標簽和所述第一深度值標注的深度訓練圖像以及具有所述第一部件分割標簽和所述第二深度值的分割訓練圖像，生成目標檢測模型訓練集；采用所述目標檢測模型訓練集，對用于深度預測和部件分割的目標檢測模型進行訓練。

根據(jù)本公開的另一方面，提供了一種目標檢測方法，包括：獲取目標檢測圖像；采用經(jīng)過訓練的目標檢測模型對所述目標檢測圖像進行檢測，以得到所述目標檢測圖像對應的部件分割標簽和深度值；其中，所述目標檢測模型，是采用本公開第一方面實施例所述的訓練方法訓練得到的。

根據(jù)本公開的另一方面，提供了一種目標檢測模型的訓練裝置，包括：獲取模塊，用于獲取標注有第一部件分割標簽的分割訓練圖像，和標注有第一深度值的深度訓練圖像；第一訓練模塊，用于采用所述標注有第一部件分割標簽的分割訓練圖像對部件分割模型進行訓練，以及采用所述標注有第一深度值的深度訓練圖像對深度估計模型進行訓練；標注模塊，用于針對每一幀所述深度訓練圖像采用經(jīng)過訓練的部件分割模型標注第二部件分割標簽，并針對每一幀分割訓練圖像采用經(jīng)過訓練的深度估計模型標注第二深度值；生成模塊，用于根據(jù)具有所述第二部件分割標簽和所述第一深度值標注的深度訓練圖像以及具有所述第一部件分割標簽和所述第二深度值的分割訓練圖像，生成目標檢測模型訓練集；第二訓練模塊，用于采用所述目標檢測模型訓練集，對用于深度預測和部件分割的目標檢測模型進行訓練。

根據(jù)本公開的另一方面，提供了一種目標檢測裝置，包括：獲取模塊，用于獲取標注有第一部件分割標簽的分割訓練圖像，和標注有第一深度值的深度訓練圖像；檢測模塊，用于采用經(jīng)過訓練的目標檢測模型對所述目標檢測圖像進行檢測，以得到所述目標檢測圖像對應的部件分割標簽和深度值；其中，所述目標檢測模型，是采用本公開第三方面實施例所述的訓練裝置訓練得到的。

根據(jù)本公開的另一方面，提供了一種電子設備，包括：至少一個處理器；以及與所述至少一個處理器通信連接的存儲器；其中，所述存儲器存儲有可被所述至少一個處理器執(zhí)行的指令，所述指令被所述至少一個處理器執(zhí)行，以使所述至少一個處理器能夠本公開第一方面實施例所述的目標檢測模型的訓練方法，或者，執(zhí)行本公開第二方面實施例所述的目標檢測方法。