[發明專利]基于風格遷移的圖像生成方法、系統、設備及存儲介質

在審

• 申請號: 202110014885.0
• 申請日: 2021-01-06
• 公開（公告）號: CN112669308A
• 公開（公告）日: 2021-04-16
• 發明（設計）人: 康睿文;羅超;成丹妮;鄒宇;李巍
• 申請（專利權）人: 攜程旅游信息技術（上海）有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T9/00;G06T5/50
• 代理公司: 上海隆天律師事務所 31282
• 代理人: 夏彬

1.一種基于風格遷移的圖像生成方法，其特征在于，包括如下步驟：

采集需要風格遷移的內容圖像和對應不同風格的風格圖像；

將所述風格圖像和所述需要風格遷移的內容圖像分別輸入圖像生成模型的編碼網絡，得到風格圖像編碼和內容圖像編碼；

將所述風格圖像編碼和所述內容圖像編碼輸入所述圖像生成模型的解碼網絡，得到風格遷移后的生成圖像。

2.根據權利要求1所述的基于風格遷移的圖像生成方法，其特征在于，將所述風格圖像編碼和所述內容圖像編碼輸入所述圖像生成模型的解碼網絡，得到風格遷移后的生成圖像，包括如下步驟：

將所述風格圖像編碼和所述內容圖像編碼輸入所述圖像生成模型的解碼器，得到編碼圖像；

將所述編碼圖像輸入所述解碼網絡的解碼層，得到風格遷移后的生成圖像。

3.根據權利要求2所述的基于風格遷移的圖像生成方法，其特征在于，將所述風格圖像編碼和所述內容圖像編碼輸入所述圖像生成模型的解碼器包括：

將各個所述內容圖像編碼分別和所對應的風格圖像編碼組合后，輸入所述圖像生成模型的解碼網絡中不同解碼器的AdaIN模塊，得到各個所述解碼器輸出的編碼圖像。

4.根據權利要求1所述的基于風格遷移的圖像生成方法，其特征在于，所述圖像生成模型為生成式對抗網絡，所述圖像生成模型包括生成器，所述生成器包括所述編碼網絡和所述解碼網絡。

5.根據權利要求1所述的基于風格遷移的圖像生成方法，其特征在于，所述采集需要風格遷移的內容圖像，包括如下步驟：

采集具有第一分辨率的原始內容圖像；

獲取所述原始內容圖像的多個具有第二分辨率的子圖像，所述第二分辨率小于所述第一分辨率；

將所述子圖像作為需要風格遷移的內容圖像。

6.根據權利要求5所述的基于風格遷移的圖像生成方法，其特征在于，所述得到風格遷移后的生成圖像之后，還包括如下步驟：

將各個所述子圖像所對應的生成圖像融合，得到所述原始內容圖像所對應的的增強圖像。

7.根據權利要求6所述的基于風格遷移的圖像生成方法，其特征在于，獲取所述原始內容圖像的多個具有第二分辨率的子圖像，包括如下步驟：

將所述原始內容圖像拆分得到多個不同移位版本的移位子圖像；

對每一個移位子圖像進行雙線性下采樣得到具有第二分辨率的子圖像。

8.根據權利要求7所述的基于風格遷移的圖像生成方法，其特征在于，將所述原始內容圖像拆分得到多個不同移位版本的移位子圖像，包括如下步驟：

確定各個移位子圖像的移位位置像素點；

對于各個移位子圖像，在其移位位置像素點處保留所述原始內容圖像對應位置的像素值，其他位置的像素值置為0，得到具有第一分辨率的移位子圖像。

9.根據權利要求8所述的基于風格遷移的圖像生成方法，其特征在于，所述將各個所述子圖像所對應的生成圖像融合，得到所述原始內容圖像所對應的的增強圖像，包括如下步驟：

根據各個所述子圖像中像素點位置與所述原始內容圖像的像素點位置的對應關系，將各個具有第二分辨率的所述生成圖像融合，得到具有第一分辨率的增強圖像。

10.根據權利要求1所述的基于風格遷移的圖像生成方法，其特征在于，所述對應不同風格的風格圖像為對應于不同光照條件的風格圖像；

所述風格遷移后的生成圖像為所述內容圖像對應于不同光照條件下的風格遷移圖像。