[發明專利]訓練數據的生成方法、試妝方法、電子設備及存儲介質

在審

• 申請號: 202111674777.2
• 申請日: 2021-12-31
• 公開（公告）號: CN114387285A
• 公開（公告）日: 2022-04-22
• 發明（設計）人: 孫宇超;金宇林
• 申請（專利權）人: 北京曠視科技有限公司;北京邁格威科技有限公司;深圳曠視金智科技有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11;G06T7/90;G06N3/04;G06N3/08
• 代理公司: 北京潤澤恒知識產權代理有限公司 11319
• 代理人: 王婷婷

1.一種訓練數據的生成方法，其特征在于，所述方法包括：

獲取第一人臉圖像和第一遮擋物圖像，其中，所述第一人臉圖像中的上妝部位未被遮擋，所述第一遮擋物圖像為用于對所述上妝部位進行遮擋的遮擋對象所對應的圖像；

獲取所述第一人臉圖像中所述上妝部位對應的第一分割掩碼圖，以及獲取所述第一遮擋物圖像中所述遮擋對象對應的第二分割掩碼圖；

使用所述第二分割掩碼圖，將所述第一人臉圖像和所述第一遮擋物圖像進行融合，生成所述上妝部位被所述遮擋對象覆蓋的第二人臉圖像；

根據所述第一分割掩碼圖和所述第二分割掩碼圖，生成所述第二人臉圖像中上妝區域對應的第三分割掩碼圖，其中，所述上妝區域包括：所述上妝部位所在區域中未被所述遮擋對象覆蓋的區域。

2.根據權利要求1所述的方法，其特征在于，在所述使用所述第二分割掩碼圖，將所述第一人臉圖像和所述第一遮擋物圖像進行融合，生成所述上妝部位被所述遮擋對象覆蓋的第二人臉圖像的步驟之前，還包括：

根據所述第一人臉圖像中皮膚區域的顏色值，對所述第一遮擋物圖像中所述遮擋對象的顏色值進行調節，得到第二遮擋物圖像，其中，所述第二遮擋物圖像中所述遮擋對象的顏色值與所述第一人臉圖像中皮膚區域的顏色值的差值小于第一數值。

3.根據權利要求2所述的方法，其特征在于，所述根據所述第一人臉圖像中皮膚區域的顏色值，對所述第一遮擋物圖像中所述遮擋對象的顏色值進行調節，得到第二遮擋物圖像，包括：

獲取所述第一人臉圖像中皮膚區域的第一平均顏色值和所述第一遮擋物圖像中所述遮擋對象所在區域的第二平均顏色值；

根據所述第一平均顏色值、所述第二平均顏色值以及所述遮擋對象所在區域中各像素點所對應的初始顏色值，確定所述遮擋對象所在區域中各像素點所對應的目標顏色值；

將所述第一遮擋物圖像中所述遮擋對象所在區域中各像素點所對應的初始顏色值替換為對應的目標顏色值，得到第二遮擋物圖像。

4.根據權利要求1-3任一項所述的方法，其特征在于，所述使用所述第二分割掩碼圖，將所述第一人臉圖像和所述第一遮擋物圖像進行融合，生成所述上妝部位被所述遮擋對象覆蓋的第二人臉圖像，包括：

對所述第二分割掩碼圖中所述遮擋對象的輪廓進行邊緣平滑處理，得到第四分割掩碼圖；

使用所述第四分割掩碼圖，將所述第一人臉圖像和所述第一遮擋物圖像進行Alpha融合，生成所述上妝部位被所述遮擋對象覆蓋的第二人臉圖像；

所述根據所述第一分割掩碼圖和所述第二分割掩碼圖，生成所述第二人臉圖像中上妝區域對應的第三分割掩碼圖，包括：

對所述第一分割掩碼圖與所述第四分割掩碼圖進行相減運算，得到所述第二人臉圖像中上妝區域對應的第三分割掩碼圖。

5.根據權利要求1-4任一項所述的方法，其特征在于，所述方法還包括：

構建初始分割模型；

將多個不同的第二人臉圖像分別輸入至所述初始分割模型進行處理，輸出對應的預測分割掩碼圖，基于所述多個不同的第二人臉圖像所對應的第三分割掩碼圖和預測分割掩碼圖，對所述初始分割模型進行訓練，直至模型收斂，得到目標分割模型。

6.一種試妝方法，其特征在于，所述方法包括：

獲取待上妝的目標人臉圖像和目標上妝產品的顏色值；

將所述目標人臉圖像輸入至目標分割模型進行處理，得到所述目標人臉圖像中目標上妝區域對應的第五分割掩碼圖，其中，所述目標分割模型是基于權利要求1-5任一項所述方法所生成的訓練數據訓練得到的；

根據目標顏色生成策略、所述目標上妝區域所對應的初始顏色值和所述目標上妝產品的顏色值，生成所述目標上妝區域所對應的目標上妝顏色值，其中，所述目標顏色生成策略包括：所述目標上妝區域中初始顏色值不高于第二數值的區域使用正片疊底的方式生成對應的目標上妝顏色值，高于所述第二數值的區域采用濾色的方式生成對應的目標上妝顏色值；

根據所述第五分割掩碼圖和所述目標上妝區域所對應的目標上妝顏色值，生成試妝人臉圖像。