[發明專利]圖像生成方法、裝置、設備及存儲介質

在審

• 申請號: 202111151607.6
• 申請日: 2021-09-29
• 公開（公告）號: CN113850212A
• 公開（公告）日: 2021-12-28
• 發明（設計）人: 劉禮杰
• 申請（專利權）人: 北京字跳網絡技術有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/34;G06K9/46;G06N3/04;G06N3/08
• 代理公司: 北京遠智匯知識產權代理有限公司 11659
• 代理人: 范坤坤

本公開實施例公開了一種圖像生成方法、裝置、設備及存儲介質。包括：獲取包含目標人體的第一人體圖像及包含目標衣物的第一衣物圖像；對所述第一人體圖像分別進行關鍵點提取、人像分割及人體部位分割，獲得關鍵點特征圖、人像分割圖及人體部位分割圖；將所述關鍵點特征圖、所述人像分割圖、所述人體部位分割圖及所述第一衣物圖像輸入形變模型中，獲得變形后的第二衣物圖像；將所述第二衣物圖像、第一人體圖像、所述關鍵點特征圖、所述人像分割圖及所述人體部位分割圖輸入混合模型，獲得第二人體圖像；其中，所述第二人體圖像中的所述目標人體穿戴所述目標衣物。本發明實施例提供的圖像生成方法，可以提高生成圖像的真實度。

技術領域

本公開實施例涉及圖像處理技術領域，尤其涉及一種圖像生成方法、裝置、設備及存儲介質。

背景技術

隨著科技的發展，越來越多的應用軟件走進了用戶的生活，逐漸豐富了用戶的業余生活，例如短視頻APP等。用戶可以采用視頻、照片等方式記錄生活，并上傳到短視頻APP上。

短視頻APP上有許多基于圖像算法與渲染技術的特效玩法。其中，虛擬換裝是指應用圖像融合技術，將用戶的人體圖像和包含目標衣物的衣物圖像進行融合，得到用戶穿戴該目標衣物后的圖像，從而在用戶無需真正試穿目標衣物的情況下，就能夠了解到目標衣物的穿戴效果。

目前，在虛擬換裝過程中，通常應用圖像融合模型，分別對人體圖像和衣物圖像進行特征提取，基于提取到的兩個圖像特征生成新的圖像，即用戶穿戴目標衣物的圖像。但是，在上述過程中，由于圖像融合模型所提取的是粗略的圖像特征，在生成圖像時容易導致新生成的圖像缺失細節信息，進而導致圖像生成效果失真，虛擬換裝的效果較差。

發明內容

本公開實施例提供一種圖像生成方法、裝置、設備及存儲介質，可以提高生成圖像的真實度。

第一方面，本公開實施例提供了一種圖像生成方法，包括：

獲取包含目標人體的第一人體圖像及包含目標衣物的第一衣物圖像；

對所述第一人體圖像分別進行關鍵點提取、人像分割及人體部位分割，獲得關鍵點特征圖、人像分割圖及人體部位分割圖；

將所述關鍵點特征圖、所述人像分割圖、所述人體部位分割圖及所述第一衣物圖像輸入形變模型中，獲得變形后的第二衣物圖像；

將所述第二衣物圖像、所述第一人體圖像、所述關鍵點特征圖、所述人像分割圖及所述人體部位分割圖輸入混合模型，獲得第二人體圖像；其中，所述第二人體圖像中的所述目標人體穿戴所述目標衣物。

第二方面，本公開實施例還提供了一種圖像生成裝置，包括：

人體圖像獲取模塊，用于獲取包含目標人體的第一人體圖像及包含目標衣物的第一衣物圖像；

分割圖獲取模塊，用于對所述第一人體圖像分別進行關鍵點提取、人像分割及人體部位分割，獲得關鍵點特征圖、人像分割圖及人體部位分割圖；

第二衣物圖像獲取模塊，用于將所述關鍵點特征圖、所述人像分割圖、所述人體部位分割圖及所述第一衣物圖像輸入形變模型中，獲得變形后的第二衣物圖像；

第二人體圖像獲取模塊，用于將所述第二衣物圖像、所述第一人體圖像、所述關鍵點特征圖、所述人像分割圖及所述人體部位分割圖輸入混合模型，獲得第二人體圖像；其中，所述第二人體圖像中的所述目標人體穿戴所述目標衣物。

第三方面，本公開實施例還提供了一種電子設備，所述電子設備包括：

一個或多個處理裝置；

存儲裝置，用于存儲一個或多個程序；

當所述一個或多個程序被所述一個或多個處理裝置執行，使得所述一個或多個處理裝置實現如本公開實施例所述的圖像生成方法。