[發明專利]數據生成方法及裝置

在審

• 申請號: 202210135524.6
• 申請日: 2022-02-14
• 公開（公告）號: CN114625343A
• 公開（公告）日: 2022-06-14
• 發明（設計）人: 付強;馬世奎
• 申請（專利權）人: 達闥機器人股份有限公司
• 主分類號: G06F8/20
• 分類號: G06F8/20;G06T17/00;G06N20/00
• 代理公司: 北京太合九思知識產權代理有限公司 11610
• 代理人: 孫明子;劉戈

本公開提供一種數據生成方法及裝置，該方法：將主體對象的屬性參數導入到游戲引擎中，以通過游戲引擎構建出對應的虛擬模型；通過游戲引擎的場景捕捉組件采集虛擬模型的圖像數據以及空間位置數據；將空間位置數據標注到圖像數據中，得到包含空間位置數據的圖像數據集。這樣，既無需借助測量設備即可獲取到虛擬模型的空間位置數據，大大簡化了空間位置數據的獲取流程，也無需人工標注即可形成包含空間位置數據的圖像數據集，大大簡化空間位置數據的標注方式，提高數據集生成效率。

技術領域

本公開涉及數據處理技術領域，尤其涉及一種數據生成方法及裝置。

背景技術

隨著機器學習的發展，機器學習的應用場景越來越廣泛。尤其是，在智能機器人領域中常會應用到機器學習模型實現各種功能。

在機器學習模型的訓練過程中，訓練數據集的質量會直接影響到機器學習模型的性能。相關技術中，通常需要依靠人工對機器學習模型的訓練數據集的部分數據進行標注。但是，機器學習模型的訓練數據集往往數據量較為龐大，常以萬為單位，導致相關技術中訓練數據集的生成過程較為繁瑣，生成效率低。

綜上，如何提高訓練數據集的生成效率，成為本公開所要解決的技術問題。

發明內容

本公開提供一種數據生成方法及裝置，用以形成包含空間位置數據的圖像數據集，提高數據集生成效率。

根據本公開實施例的第一方面，本公開提供一種數據生成方法，包括：

將主體對象的屬性參數導入到游戲引擎中，以通過游戲引擎構建出對應的虛擬模型；

通過游戲引擎的場景捕捉組件，采集虛擬模型的圖像數據以及空間位置數據；

將空間位置數據標注到圖像數據中，得到包含空間位置數據的圖像數據集。

根據本公開實施例的第二方面，本公開提供一種數據生成裝置，包括：

導入模塊，被配置為將主體對象的屬性參數導入到游戲引擎中，以通過游戲引擎構建出對應的虛擬模型；

采集模塊，被配置為通過游戲引擎的場景捕捉組件，采集虛擬模型的圖像數據以及空間位置數據；

標注模塊，被配置為將空間位置數據標注到圖像數據中，得到包含空間位置數據的圖像數據集。

根據本公開實施例的第三方面，提供一種電子設備，其中包括處理器和存儲器，其中，存儲器上存儲有可執行代碼，當可執行代碼被處理器執行時，使處理器至少可以實現第一方面中的數據生成方法。

根據本公開實施例的第四方面，提供了一種計算機可讀存儲介質，當所述計算機可讀存儲介質中的指令由電子設備執行時，使得電子設備能夠執行至少可以實現第一方面中的數據生成方法。

根據本公開實施例的第五方面，提供一種計算機程序產品，包括計算機程序/指令，所述計算機程序/指令被處理器執行時實現第一方面中的數據生成方法。

本公開的實施例提供的技術方案至少帶來以下有益效果：

本公開中，針對需要標注的主體對象，將主體對象的屬性參數導入到游戲引擎中，以通過游戲引擎構建出對應的虛擬模型，進而通過游戲引擎的場景捕捉組件采集虛擬模型的圖像數據以及空間位置數據。這樣，無需借助測量設備即可獲取到虛擬模型的空間位置數據，大大簡化了空間位置數據的獲取流程。進而，將空間位置數據標注到圖像數據中，得到包含空間位置數據的圖像數據集，從而無需人工標注即可形成包含空間位置數據的圖像數據集，大大簡化空間位置數據的標注方式，提高數據集生成效率。

附圖說明

此處的附圖被并入說明書中并構成本說明書的一部分，示出了符合本公開的實施例，并與說明書一起用于解釋本公開的原理，并不構成對本公開的不當限定。