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

在審

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

1.一種模型訓練數據獲取方法，其特征在于，所述方法包括：

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

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

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

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

2.根據權利要求1所述的方法，其特征在于，所述場景配置參數包括所述虛擬相機的相機位姿參數、所述虛擬仿真場景中目標物體的物體位姿參數以及環境參數；

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

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

3.根據權利要求1所述的方法，其特征在于，所述基于掛載于所述虛擬相機上的可執行程序，對所述虛擬仿真場景的場景配置參數進行多次調整，并通過所述虛擬相機采集每一次所述場景配置參數調整后的虛擬仿真場景的圖像數據，得到用于對網絡模型進行訓練的初始訓練圖像，包括：

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

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

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

4.根據權利要求3所述的方法，其特征在于，所述對所述初始虛擬仿真場景的場景配置參數進行多次調整，得到各場景配置參數調整后的虛擬仿真場景，并采集各所述場景配置參數調整后的虛擬仿真場景的圖像數據，得到第二圖像數據，包括：

對所述初始虛擬仿真場景的場景配置參數進行調整，得到場景配置參數調整后的虛擬仿真場景，并在確定采集得到所述場景配置參數調整后的虛擬仿真場景的圖像數據后，再進行下一次場景配置參數的調整。

5.根據權利要求3所述的方法，其特征在于，所述對所述初始虛擬仿真場景的場景配置參數進行多次調整，得到各場景配置參數調整后的虛擬仿真場景，并采集各所述場景配置參數調整后的虛擬仿真場景的圖像數據，得到第二圖像數據，包括：

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

記錄對所述場景配置參數進行調整的調整次數，并判斷所述調整次數是否等于預設閾值，所述預設閾值基于網絡模型的訓練要求確定；

當確定所述調整次數等于所述預設閾值時，停止對所述場景配置參數繼續進行調整，得到對應各場景配置參數調整后的虛擬仿真場景的第二圖像數據；

當確定所述調整次數小于所述預設閾值時，繼續對所述場景配置參數進行調整，直至所述調整次數等于所述預設閾值。