[發明專利]數據生成方法、裝置及電子設備

有效

• 申請號: 202110431972.6
• 申請日: 2021-04-21
• 公開（公告）號: CN113269782B
• 公開（公告）日: 2023-01-03
• 發明（設計）人: 吳濤
• 申請（專利權）人: 青島小鳥看看科技有限公司
• 主分類號: G06T7/10
• 分類號: G06T7/10;G06T19/00;G06N3/04;G06N3/08
• 代理公司: 北京博雅睿泉專利代理事務所(特殊普通合伙) 11442
• 代理人: 李永凱

1.一種數據生成方法，包括：

獲取第一圖像數據，其中，所述第一圖像數據為表示用戶所處真實環境的數據；

獲取目標對象的類別信息和平面信息，其中，所述目標對象為所述第一圖像數據中的對象，所述類別信息為表示所述目標對象的物體類型的信息，所述平面信息包括所述目標對象的外表面的信息；

獲取第二圖像數據，其中，所述第二圖像數據為包含虛擬對象的數據；

根據所述類別信息和所述平面信息，混合所述第一圖像數據和所述第二圖像數據，生成目標圖像數據，其中，所述目標圖像數據為包含所述目標對象和所述虛擬對象的數據，

其中，所述根據所述平面信息和所述類別信息，混合所述第一圖像數據和所述第二圖像數據，生成目標圖像數據，包括：

根據所述類別信息，確定所述第二圖像數據中的所述虛擬對象與所述第一圖像數據中的所述目標對象之間的相對位置關系；

根據所述平面信息和所述相對位置關系，將所述虛擬對象渲染至所述目標對象的預設位置處，獲得所述目標圖像數據。

2.根據權利要求1所述的方法，所述獲取所述目標對象的類別信息和平面信息，包括：

將所述第一圖像數據輸入到目標圖像分割模型中，獲得所述目標對象的掩膜信息；

根據所述掩膜信息，獲得所述類別信息和所述平面信息。

3.根據權利要求2所述的方法，所述根據所述掩膜信息，獲得所述類別信息，包括：

將所述掩膜信息輸入到目標類別識別模型中，獲得所述類別信息。

4.根據權利要求2所述的方法，所述根據所述掩膜信息，獲得所述平面信息，包括：

根據所述掩膜信息，獲得所述目標對象在所述第一圖像數據中對應的目標圖像塊；

根據所述目標圖像塊，獲取所述目標對象的關鍵點在世界坐標系下的目標位置信息，其中，所述關鍵點包括所述目標對象的角點；

根據所述目標位置信息和預設平面擬合算法，獲得所述平面信息，其中，所述平面信息包括與所述目標對象的每一平面對應的中心點坐標和平面法向量。

5.根據權利要求4所述的方法，所述方法應用于電子設備，所述根據所述目標圖像塊，獲取所述目標對象的關鍵點在世界坐標系下的目標位置信息，包括：

根據所述目標圖像塊，檢測所述關鍵點在所述第一圖像數據中的第一位置信息；

獲取所述電子設備在第一時刻的位姿信息，以及，所述關鍵點在第二時刻獲取到的第三圖像數據中的第二位置信息，其中，所述第一時刻包括當前時刻，所述第二時刻早于所述第一時刻；

根據所述第一位置信息、所述位姿信息和所述第二位置信息，獲得所述目標位置信息。

6.根據權利要求3所述的方法，所述目標圖像分割模型和所述目標類別識別模型通過以下步驟訓練獲得：

獲取樣本數據，其中，所述樣本數據為包含預設場景中的樣本對象的數據；

根據所述樣本數據，聯合訓練初始圖像分割模型和初始類別識別模型，獲得所述目標圖像分割模型和所述目標類別識別模型。

7.根據權利要求1所述的方法，在獲得所述目標圖像數據之后，所述方法還包括：

展示所述目標圖像數據。