[發明專利]一種圖像處理方法、裝置、電子設備及存儲介質

在審

• 申請號: 201910918266.7
• 申請日: 2019-09-26
• 公開（公告）號: CN110705625A
• 公開（公告）日: 2020-01-17
• 發明（設計）人: 郭曉鋒;姚曉宇;譚穎;李海
• 申請（專利權）人: 北京奇藝世紀科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/00
• 代理公司: 11662 北京華夏泰和知識產權代理有限公司
• 代理人: 郭金鑫

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

獲取訓練樣本集，其中，所述訓練樣本集包括源樣本圖像以及目標樣本圖像；

根據所述訓練樣本集訓練圖像轉換模型和第一判別模型，所述源樣本圖像經過所述圖像轉換模型處理后得到目標圖像；

根據所述第一判別模型計算所述目標圖像和所述目標樣本圖像之間的所述第一對抗損失；

當所述第一對抗損失符合第一預設條件時，得到第一目標圖像轉換模型。

2.根據權利要求1所述的方法，其特征在于，所述方法還包括：

當所述第一對抗損失不符合第一預設條件時，根據所述第一對抗損失對所述圖像轉換模型和第一判別模型中的參數進行調整；

所述第一對抗損失符合第一預設條件包括：所述第一對抗損失小于或等于第一預設閾值，或所述第一對抗損失不再變化。

3.根據權利要求1所述的方法，其特征在于，所述方法還包括：

根據所述訓練樣本集訓練圖像逆轉換模型和第二判別模型，所述目標樣本圖像經過所述圖像逆轉換模型處理后得到第一源重構圖像；

根據所述第二判別模型計算所述第一源重構圖像和所述源樣本圖像的第二對抗損失；

基于所述第一對抗損失和第二對抗損失計算最終對抗損失；

當所述最終對抗損失不符合第二預設條件時，基于所述第一對抗損失和第二對抗損失對所述圖像轉換模型、圖像逆轉換模型、第一判別模型和第二判別模型中的參數進行調整，直至所述最終對抗損失符合所述第二預設條件，得到第二目標圖像轉換模型；

其中，所述最終對抗損失符合所述第二預設條件包括：所述最終對抗損失小于或等于第二預設閾值，或所述最終對抗損失不再變化。

4.根據權利要求3所述的方法，其特征在于，所述方法還包括：

根據所述訓練樣本集訓練圖像轉換模型和圖像逆轉換模型，根據所述圖像逆轉換模型將所述目標圖像轉換為第二源重構圖像；

根據所述圖像轉換模型將所述第一源重構圖像轉換為目標重構圖像；

計算所述源樣本圖像與所述第二源重構圖像之間的第一重構誤差；

計算所述目標樣本圖像與所述目標重構圖像之間的第二重構誤差；

根據所述第一重構誤差和第二重構誤差計算最終重構誤差；

當所述最終重構誤差不符合第三預設條件時，基于所述第一重構誤差和第二重構誤差對所述圖像轉換模型和圖像逆轉換模型中的參數進行調整，直至所述最終重構誤差符合所述第三預設條件，得到第三目標圖像轉換模型；

其中，所述最終重構誤差符合所述第三預設條件包括：所述最終重構誤差小于或等于第三預設閾值，或所述最終重構誤差不再變化。

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

基于所述第一對抗損失、第二對抗損失、第一重構誤差和第二重構誤差計算最終損失；

當所述最終損失不符合第四預設條件時，基于所述第一對抗損失、第二對抗損失、第一重構誤差和第二重構誤差對所述圖像轉換模型和圖像逆轉換模型中的參數進行調整，直至所述最終損失符合所述第四預設條件，得到第四目標圖像轉換模型；

其中，所述最終損失符合所述第四預設條件包括：所述最終損失小于或等于第四預設閾值，或所述最終損失不再變化。

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

獲取源風格臉部圖像；

根據圖像轉換模型將所述源風格臉部圖像轉換為目標風格臉部圖像。

7.根據權利要求6所述的方法，其特征在于，所述根據圖像轉換模型將所述源風格臉部圖像轉換為目標風格臉部圖像，包括：

提取所述源風格臉部圖像的第一臉部特征信息；

將所述第一臉部特征信息轉換為目標風格的第二臉部特征信息；

根據所述第二臉部特征信息生成所述目標風格臉部圖像。