[發明專利]一種圖像處理方法、裝置、設備及計算機可讀存儲介質

有效

• 申請號: 202110620382.8
• 申請日: 2021-06-03
• 公開（公告）號: CN113822790B
• 公開（公告）日: 2023-04-21
• 發明（設計）人: 朱飛達;邰穎;汪鋮杰;李季檁
• 申請（專利權）人: 騰訊云計算（北京）有限責任公司
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06V10/774;G06V10/40
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 熊永強;杜維

1.一種圖像處理方法，其特征在于，包括：

獲取圖像處理模型的訓練樣本集，所述圖像處理模型包含第一生成網絡和第二生成網絡，所述第一生成網絡和所述第二生成網絡具備相同的特征提取結構；所述訓練樣本集包含第一樣本集和第二樣本集，所述第一樣本集包括N個第一樣本圖像，每個第一樣本圖像均包含目標頭像；所述第二樣本集包括M個第二樣本圖像，每個第二樣本圖像包含一個樣本頭像，M、N均為正整數；

從所述第一樣本集中選取第一樣本圖像x_i，調用所述第一生成網絡對所述第一樣本圖像x_i進行生成處理，得到第一預測圖像x′_i；所述第一預測圖像x′_i包含目標頭像，且所述第一預測圖像x′_i中的目標頭像的屬性與所述第一樣本圖像x_i中的目標頭像的屬性保持一致，i為正整數且i≤N；

從所述第二樣本集中選取第二樣本圖像y_k，調用所述第一生成網絡對所述第二樣本圖像y_k進行生成處理，得到第二預測圖像y′_k；其中，所述第二預測圖像y′_k包含目標頭像，且所述第二預測圖像y′_k中的目標頭像的屬性與所述第二樣本圖像y_k中的樣本頭像的屬性保持一致，k為正整數且k≤M；

調用所述第二生成網絡對所述第二樣本圖像y_k進行生成處理，得到第三預測圖像y″_k；所述第三預測圖像y″_k與所述第二樣本圖像y_k包含相同的樣本頭像；

根據所述第一樣本圖像x_i與所述第一預測圖像x′_i之間的差異，所述第二樣本圖像y_k與所述第二預測圖像y′_k之間的差異，以及所述第二樣本圖像y_k與所述第三預測圖像y″_k之間的差異，更新所述圖像處理模型的參數；

令i＝i+1且k＝k+1，迭代上述訓練流程直至所述圖像處理模型達到收斂，其中，i+1≤N，k+1≤M。

2.如權利要求1所述的方法，其特征在于，所述第一生成網絡包括編碼器和第一解碼器；所述調用所述第一生成網絡對所述第一樣本圖像x_i進行生成處理，得到第一預測圖像x′_i，包括：

調用所述編碼器對所述第一樣本圖像x_i進行編碼處理，得到第一特征向量，所述第一特征向量蘊含所述第一樣本圖像x_i中的目標頭像的屬性；

調用所述第一解碼器對所述第一特征向量進行解碼處理，得到第一生成圖像及所述第一生成圖像的區域分割信息，所述區域分割信息用于指示所述第一生成圖像中的頭像區域；

按照所述第一生成圖像的區域分割信息，從所述第一生成圖像中提取第一預測圖像x′_i。

3.如權利要求1所述的方法，其特征在于，所述第一生成網絡包括編碼器和第一解碼器；所述調用所述第一生成網絡對所述第二樣本圖像y_k進行生成處理，得到第二預測圖像y′_k，包括：

調用所述編碼器對所述第二樣本圖像y_k進行編碼處理，得到第二特征向量，所述第二特征向量蘊含所述第二樣本圖像y_k中的樣本頭像的屬性；

調用所述第一解碼器對所述第二特征向量進行解碼處理，得到第二生成圖像及所述第二生成圖像的區域分割信息，所述區域分割信息用于指示所述第二生成圖像中的頭像區域；

按照所述第二生成圖像的區域分割信息，從所述第二生成圖像中提取第二預測圖像y′_k。