[發明專利]模型訓練數據獲取方法、裝置、計算機設備和存儲介質

在審

• 申請號: 202110397798.8
• 申請日: 2021-04-14
• 公開（公告）號: CN113516778A
• 公開（公告）日: 2021-10-19
• 發明（設計）人: 盧志巍
• 申請（專利權）人: 武漢聯影智融醫療科技有限公司
• 主分類號: G06T19/00
• 分類號: G06T19/00;G06T7/80
• 代理公司: 北京華進京聯知識產權代理有限公司 11606
• 代理人: 喬改利

本申請涉及人工智能技術領域，特別是涉及一種模型訓練數據獲取方法、裝置、計算機設備和存儲介質。所述方法包括：構建對應真實場景的虛擬仿真場景，虛擬仿真場景中包括真實場景中對應的各實體物體以及虛擬相機；基于掛載于虛擬相機上的可執行程序，對虛擬仿真場景的場景配置參數進行多次調整，并通過虛擬相機采集每一次場景配置參數調整后的虛擬仿真場景的圖像數據，得到用于對網絡模型進行訓練的初始訓練圖像；對初始訓練圖像進行識別，確定初始訓練圖像中的目標物體；基于虛擬仿真場景的模型數據，對初始訓練圖像中的目標物體進行標注，得到標注后的用于對網絡模型進行訓練的訓練數據。采用本方法能夠提升模型訓練數據生成智能化水平。

技術領域

本申請涉及人工智能技術領域，特別是涉及一種模型訓練數據獲取方法、裝置、計算機設備和存儲介質。

背景技術

隨著人工智能在圖像處理技術上的不斷發展，計算機設備能夠利用如機器學習技術，對圖像或者視頻進行個性化處理，如圖像分類、目標檢測、圖像分割、關鍵點檢測等，得到滿足用戶實際需求的圖像處理結果。

在傳統方式中，為了提高圖像處理效率及準確性，通常是利用機器學習網絡(尤其是深度學習網絡)，對大量帶有標注信息的樣本數據進行有監督訓練，得到滿足相應場景的圖像處理任務需求的圖像處理模型。在該種方式中，需要從真實場景中采集大量的數據，并對采集的數據通過人工進行標注，以得到訓練數據。

通過人工進行訓練數據的標注，使得訓練數據的生成處理過程智能化水平較低。

發明內容

基于此，有必要針對上述技術問題，提供一種能夠提升模型訓練數據生成智能化水平的模型訓練數據獲取方法、裝置、計算機設備和存儲介質。

一種模型訓練數據獲取方法，所述方法包括：

構建對應真實場景的虛擬仿真場景，虛擬仿真場景中包括真實場景中對應的各實體物體以及虛擬相機；

基于掛載于虛擬相機上的可執行程序，對虛擬仿真場景的場景配置參數進行多次調整，并通過虛擬相機采集每一次場景配置參數調整后的虛擬仿真場景的圖像數據，得到用于對網絡模型進行訓練的初始訓練圖像；

根據虛擬仿真場景的模型數據，對初始訓練圖像進行識別，得到初始訓練圖像中的目標物體的識別結果；

基于識別結果，對初始訓練圖像中的目標物體進行標注，得到標注后的用于對網絡模型進行訓練的訓練數據。

在其中一個實施例中，場景配置參數包括虛擬相機的相機位姿參數、虛擬仿真場景中目標物體的物體位姿參數以及環境參數；

基于掛載于虛擬相機上的可執行程序，對虛擬仿真場景的場景配置參數進行多次調整，并通過虛擬相機采集每一次場景配置參數調整后的虛擬仿真場景的圖像數據，得到用于對網絡模型進行訓練的初始訓練圖像，包括：

基于掛載于虛擬相機上的可執行程序，對虛擬仿真場景的相機位姿參數、物體位姿參數以及環境參數至少一項進行調整，并通過虛擬相機采集每一次相機位姿參數、物體位姿參數以及環境參數至少一項調整后的虛擬仿真場景的圖像數據，得到用于對網絡模型進行訓練的初始訓練圖像。

在其中一個實施例中，基于掛載于虛擬相機上的可執行程序，對虛擬仿真場景的場景配置參數進行多次調整，并通過虛擬相機采集每一次場景配置參數調整后的虛擬仿真場景的圖像數據，得到用于對網絡模型進行訓練的初始訓練圖像，包括：

基于掛載于虛擬相機上的可執行程序，采集構建的初始虛擬仿真場景的圖像數據，得到第一圖像數據；

對初始虛擬仿真場景的場景配置參數進行多次調整，得到各場景配置參數調整后的虛擬仿真場景，并采集各場景配置參數調整后的虛擬仿真場景的圖像數據，得到第二圖像數據；

根據第一圖像數據以及第二圖像數據，得到對網絡模型進行訓練的初始訓練圖像。