[發明專利]圖像色偏校正方法、模型訓練方法、裝置及設備

在審

• 申請號: 202011385880.0
• 申請日: 2020-12-01
• 公開（公告）號: CN112508812A
• 公開（公告）日: 2021-03-16
• 發明（設計）人: 陳揚;李啟東;李志陽;周銘柯;陳進山
• 申請（專利權）人: 廈門美圖之家科技有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 榮穎佳

本申請提供一種圖像色偏校正方法、模型訓練方法、裝置及設備，屬于圖像處理技術領域。圖像色偏校正方法，包括：獲取待校正圖像；將待校正圖像輸入預先訓練得到的色偏校正網絡模型中，由色偏校正網絡模型進行校正處理，得到校正后的圖像，其中，色偏校正網絡模型為卷積神經網絡模型，色偏校正網絡模型包括：編碼網絡、解碼網絡以及全局特征提取網絡，全局特征提取網絡用于基于編碼網絡的下采樣結果得到待校正圖像的全局特征，解碼網絡用于基于待校正圖像的全局特征以及編碼網絡的下采樣結果進行解碼處理。本申請可以提高色偏校正的效果。

技術領域

本申請涉及圖像處理技術領域，具體而言，涉及一種圖像色偏校正方法、模型訓練方法、裝置及設備。

背景技術

在圖像處理技術領域中，由于拍照時的光線等環境影響因素，可能會導致拍攝得到的照片的光線較亮或者較暗，不能夠準確反映出拍攝物體或者場景的真實顏色，也即是說這些照片上的圖像出現了顏色偏差。

為了對這類出現顏色偏差的圖像進行校正，現有技術中通常是采用白平衡處理的方式來進行校正，然而現有的白平衡處理中并不能徹底校正色偏，會導致的色偏校正的效果較差。

發明內容

本申請的目的在于提供一種圖像色偏校正方法、模型訓練方法、裝置及設備，可以提高色偏校正的效果。

本申請的實施例是這樣實現的：

本申請實施例的一方面，提供一種圖像色偏校正方法，包括：

獲取待校正圖像；

將待校正圖像輸入預先訓練得到的色偏校正網絡模型中，由色偏校正網絡模型進行校正處理，得到校正后的圖像，其中，色偏校正網絡模型為卷積神經網絡模型，色偏校正網絡模型包括：編碼網絡、解碼網絡以及全局特征提取網絡，全局特征提取網絡用于基于編碼網絡的下采樣結果得到待校正圖像的全局特征，解碼網絡用于基于待校正圖像的全局特征以及編碼網絡的下采樣結果進行解碼處理。

可選地，將待校正圖像輸入預先訓練得到的色偏校正網絡模型中，由色偏校正網絡模型進行校正處理，得到校正后的圖像，包括：

將待校正圖像輸入編碼網絡中，由編碼網絡對待校正圖像進行多次下采樣處理，得到下采樣處理結果；

將下采樣處理結果輸入全局特征提取網絡中，由全局特征提取網絡基于下采樣處理結果提取得到待校正圖像的全局特征；

將下采樣結果以及待校正圖像的全局特征輸入解碼網絡，由解碼網絡基于下采樣結果以及待校正圖像的全局特征進行多次上采樣處理，得到校正后的圖像。

可選地，將下采樣處理結果輸入全局特征提取網絡中，由全局特征提取網絡基于下采樣處理結果提取得到待校正圖像的全局特征，包括：

將下采樣處理結果輸入全局特征提取網絡中，由全局特征提取網絡對下采樣處理結果依次進行均值池化、全連接以及擴充處理，得到待校正圖像的全局特征。

可選地，全局特征提取網絡包括：均值池化層、依次連接的多個全連接層以及擴充處理層。

可選地，獲取待校正圖像之前，還包括：

對初始圖像進行校正預處理，得到初始圖像對應的待校正圖像，校正預處理包括下采樣預處理以及場景預處理。

可選地，對初始圖像進行校正預處理，得到初始圖像對應的待校正圖像，包括：

對初始圖像進行下采樣預處理，得到預設尺寸的圖像，并將預設尺寸的圖像作為待校正圖像；

對預設尺寸的圖像進行場景預處理，以獲取場景融合系數。

可選地，將待校正圖像輸入預先訓練得到的色偏校正網絡模型中，由色偏校正網絡模型進行校正處理，得到校正后的圖像之后，還包括：