[發明專利]圖像處理方法、遷移模型訓練方法、裝置、介質及設備

有效

• 申請號: 202010628695.3
• 申請日: 2020-07-01
• 公開（公告）號: CN111784566B
• 公開（公告）日: 2022-02-08
• 發明（設計）人: 王一同;朱淵略
• 申請（專利權）人: 北京字節跳動網絡技術有限公司
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06T3/40;G06T7/90;G06N3/04;G06N3/08
• 代理公司: 北京英創嘉友知識產權代理事務所(普通合伙) 11447
• 代理人: 賈會玲

本公開涉及一種圖像處理方法、遷移模型訓練方法、裝置、介質及設備。方法包括：獲取待處理圖像、具有第一圖像風格的第一目標圖像和具有第二圖像風格的第二目標圖像，第一圖像風格包括第一顏色風格和第一形狀風格，第二圖像風格包括第二顏色風格和第二形狀風格；將待處理圖像、第一目標圖像和第二目標圖像輸入圖像風格遷移模型，以按照目標顏色風格和目標形狀風格對待處理圖像進行風格遷移處理，目標顏色風格為第一顏色風格，目標形狀風格根據第一形狀風格和第二形狀風格確定，或者，目標顏色風格根據第一顏色風格和第二顏色風格確定，目標形狀風格為第一形狀風格。由此，可實現生成圖像的顏色風格或形狀風格的獨立控制，以滿足用戶的多樣化需求。

技術領域

本公開涉及圖像處理領域，具體地，涉及一種圖像處理方法、遷移模型訓練方法、裝置、介質及設備。

背景技術

圖像處理是一種具有巨大的社會和經濟效益的實用技術，被廣泛應用于各行各業以及人們日常生活中。圖像處理中最常見的一個需求是改變圖像風格(即風格遷移)，其目標是保持待處理圖像的內容信息(例如，人臉、動物等)同時將一張風格圖像的圖像風格遷移至該待處理圖像中。例如，將真人頭像變成具有動漫風格的頭像(即動漫頭像)。

圖像風格包括顏色風格(例如，動漫頭像中頭發、眼睛的顏色)和形狀風格(例如，動漫頭像中嘴巴和眼睛的形狀)。通常風格遷移通過給定的一張待處理圖像和一張風格圖像，生成具有待處理圖像的內容信息和風格圖像的顏色風格、形狀風格的圖像。在實際使用中，風格遷移也可以通過給定的一種待處理圖像和兩張風格圖像，生成具有待處理圖像的內容信息和兩張風格圖像的顏色風格融合后的顏色風格、兩張風格圖像的形狀風格融合后的形狀風格。可見，生成的圖像的顏色風格為兩張風格圖像的顏色風格的融合、形狀風格為兩張風格圖像的形狀風格的融合，而無法使得生成的圖像的顏色風格與其中一張風格圖像的顏色風格相同，而形狀風格為兩張風格圖像的形狀風格的融合，或者，生成的圖像的形狀風格與其中一張風格圖像的形狀風格相同，而顏色風格為兩張風格圖像的顏色風格的融合。

發明內容

提供該發明內容部分以便以簡要的形式介紹構思，這些構思將在后面的具體實施方式部分被詳細描述。該發明內容部分并不旨在標識要求保護的技術方案的關鍵特征或必要特征，也不旨在用于限制所要求的保護的技術方案的范圍。

第一方面，本公開提供一種圖像處理方法，包括：獲取待處理圖像、第一目標圖像和第二目標圖像，其中，所述第一目標圖像具有第一圖像風格，所述第二目標圖像具有第二圖像風格，所述第一圖像風格包括第一顏色風格和第一形狀風格，所述第二圖像風格包括第二顏色風格和第二形狀風格；將所述待處理圖像、所述第一目標圖像和所述第二目標圖像輸入至圖像風格遷移模型中，以通過所述圖像風格遷移模型按照目標顏色風格和目標形狀風格，對所述待處理圖像進行風格遷移處理，其中，所述目標顏色風格為所述第一顏色風格，而所述目標形狀風格是根據所述第一形狀風格和所述第二形狀風格確定的，或者，所述目標顏色風格是根據所述第一顏色風格和所述第二顏色風格確定的，而所述目標形狀風格為所述第一形狀風格。