[發明專利]圖像色彩遷移方法及裝置、計算機可讀介質和電子設備

在審

• 申請號: 202110566218.3
• 申請日: 2021-05-24
• 公開（公告）號: CN113298731A
• 公開（公告）日: 2021-08-24
• 發明（設計）人: 汪路超
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T7/90
• 代理公司: 深圳市隆天聯鼎知識產權代理有限公司 44232
• 代理人: 劉抗美

1.一種圖像色彩遷移方法，其特征在于，包括：

獲取內容圖像以及參考圖像；

將所述內容圖像和所述參考圖像進行參數化處理，得到所述內容圖像對應的第一統計特征和所述參考圖像的第二統計特征；

對所述第一統計特征和所述第二統計特征進行迭代優化，得到所述內容圖像和所述參考圖像之間的色彩映射關系；

根據所述色彩映射關系對所述內容圖像進行色彩遷移處理，得到與所述參考圖像具有相似色彩分布特征的目標內容圖像。

2.根據權利要求1所述的方法，其特征在于，所述統計特征包括概率分布；所述將所述內容圖像和所述參考圖像進行參數化處理，得到所述內容圖像對應的第一統計特征和所述參考圖像的第二統計特征，包括：

基于高斯混合模型對所述內容圖像進行混合高斯擬合，得到所述內容圖像的第一概率分布；以及

基于高斯混合模型對所述參考圖像進行混合高斯擬合，得到所述參考圖像的第二概率分布。

3.根據權利要求1所述的方法，其特征在于，對所述第一統計特征和所述第二統計特征進行迭代優化，得到所述內容圖像和所述參考圖像之間的色彩映射關系，包括：

構建分布擬合模型，所述分布擬合模型為卷積神經網絡模型，所述分布擬合模型的損失函數為相對熵損失函數；

基于所述分布擬合模型對所述第一統計特征和所述第二統計特征進行非線性優化，以最小化所述相對熵損失函數，得到所述內容圖像和所述參考圖像之間的色彩映射關系。

4.根據權利要求1所述的方法，其特征在于，根據所述色彩映射關系對所述內容圖像進行色彩遷移處理，得到與所述參考圖像具有相似色彩分布特征的目標內容圖像，包括：

根據所述色彩映射關系將所述內容圖像中的原始像素值轉化為目標像素值；

通過所述目標像素值生成與所述參考圖像具有相似色彩分布特征的目標內容圖像。

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

將所述內容圖像、所述參考圖像和所述目標內容圖像作為樣本圖像集合；

基于所述樣本圖像集合對預構建的色彩遷移模型進行有監督訓練，得到訓練完成的色彩遷移模型。

6.根據權利要求5所述的方法，其特征在于，所述色彩遷移模型包括卷積層，所述卷積層包括1*1卷積核；所述方法還包括：

將所述內容圖像和所述參考圖像輸入到所述預構建的色彩遷移模型中，以通過所述1*1卷積核分別提取所述內容圖像中第一圖像特征的和所述參考圖像中的第二圖像特征。

7.根據權利要求6所述的方法，其特征在于，所述色彩遷移模型包括自適應示例歸一化模塊；所述方法還包括：

通過所述自適應示例歸一化模塊將所述第一圖像特征的均值和方差對齊到所述第二圖像特征的均值和方差，以實現將所述參考圖像中的色彩分布特征遷移到所述內容圖像中。

8.一種圖像色彩遷移裝置，其特征在于，包括：

圖像獲取模塊，用于獲取內容圖像以及參考圖像；

圖像參數化模塊，用于將所述內容圖像和所述參考圖像進行參數化處理，得到所述內容圖像對應的第一統計特征和所述參考圖像的第二統計特征；

優化擬合模塊，用于對所述第一統計特征和所述第二統計特征進行迭代優化，得到所述內容圖像和所述參考圖像之間的色彩映射關系；

圖像色彩遷移模塊，用于根據所述色彩映射關系對所述內容圖像進行色彩遷移處理，得到與所述參考圖像具有相似色彩分布特征的目標內容圖像。

9.一種計算機可讀介質，其上存儲有計算機程序，其特征在于，所述計算機程序被處理器執行時實現如權利要求1至7中任一項所述的方法。

10.一種電子設備，其特征在于，包括：

處理器；以及

存儲器，用于存儲所述處理器的可執行指令；

其中，所述處理器配置為經由執行所述可執行指令來執行權利要求1至7任一項所述的方法。