[發明專利]一種圖像白平衡校正方法及相關裝置

在審

• 申請號: 202210336143.4
• 申請日: 2022-03-31
• 公開（公告）號: CN114745531A
• 公開（公告）日: 2022-07-12
• 發明（設計）人: 張英強;周桐;王瓊瑤;劉新宇;趙磊
• 申請（專利權）人: 北京紫光展銳通信技術有限公司
• 主分類號: H04N9/73
• 分類號: H04N9/73;H04N9/64
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 夏秋

本申請實施例公開一種圖像白平衡校正方法及相關裝置。其中，圖像白平衡校正方法包括：獲取待校準圖像；基于N種預設環境光源對該待校準圖像進行渲染，得到N張第一圖像；其中，第一圖像與預設環境光源一一對應，該N為大于1的整數；將該N張第一圖像輸入深度學習白平衡校正模型，得到該待校準圖像對應的白平衡校正圖像。通過這樣的方式，無須預先進行混合色溫場景的判斷，從而簡化對圖像進行白平衡處理的流程；并且在對混合色溫圖像進行白平衡處理的過程中不會引入對環境光源的估計誤差，提升了白平衡處理的準確度。

技術領域

本申請涉及信息處理領域，尤其涉及一種圖像白平衡校正方法及相關裝置。

背景技術

為了解決成像偏色問題，減少環境光源顏色的影響，在攝像頭成像時盡可能準確的反映物體本身的顏色，提出了白平衡校正算法。傳統的白平衡校正算法在單一色溫光源場景下的圖像(后文簡稱單色溫圖像)可以取得較好的效果。但在實際情況下，通常會出現兩種及以上的色溫光源(后文簡稱混合色溫場景)，在這種混合色溫場景中，傳統的白平衡校正算法難以取得較好的效果。

為了解決混合色溫場景下傳統的白平衡校正算法效果較差的問題，通常在判斷待處理圖像為混合色溫場景下的圖像之后，對混合色溫場景的多個環境光源分別進行估計，并根據該估計結果對混合色溫場景下的每種色溫場景分別進行白平衡處理。

但通過這樣的白平衡處理方法，有可能引入對環境光源估計的誤差，從而降低對混合色溫場景下的圖像(后文簡稱混合色溫圖像)白平衡處理的準確度?？梢?，如何提升混合色溫場景下的圖像白平衡處理的準確度是一個亟待解決的問題。

發明內容

本申請實施例提供了一種圖像白平衡校正方法及相關裝置，通過本申請提供的方法，在對混合色溫圖像進行白平衡處理的過程中，不會引入對環境光源的估計誤差，且通過深度學習模型進行深層語義特征學習和白平衡處理，可以提升對混合色溫圖像白平衡處理的準確度。

第一方面，本申請實施例提供了一種圖像白平衡校正方法，該方法包括：

獲取待校準圖像；基于N種預設環境光源對待校準圖像進行渲染，得到N張第一圖像；其中，第一圖像與預設環境光源一一對應，N為大于1的整數；將N張第一圖像輸入深度學習白平衡校正模型，得到待校準圖像對應的白平衡校正圖像。

可見，通過該圖像白平衡校正方法對混合色溫圖像進行校正，可以避免引入對混合色溫環境光源的估計誤差，同時基于深度學習模型進行深層語義特征提取和白平衡校正，可以顯著提升對混合色溫圖像白平衡處理的準確度。

一種可能的實現方式中，該待校準圖像包括單色溫圖像或混合色溫圖像。通過實施該可能的實施方式，可見在對待校準圖像進行白平衡處理之前，無須預先進行混合色溫場景的判斷，從而簡化對圖像進行白平衡處理的流程。

一種可能的實現方式中，將N張第一圖像輸入深度學習白平衡校正模型，得到N個第一權重圖像；其中，第一權重圖像與第一圖像一一對應；基于N個第一權重圖像對N張第一圖像進行融合處理，得到待校準圖像對應的白平衡校正圖像。

一種可能的實現方式中，獲取多張單色溫圖像中每張單色溫圖像對應的白平衡標簽圖像；基于M種預設環境光源對每張白平衡標簽圖像進行渲染，得到每張白平衡標簽圖像對應的混合色溫圖像；其中，M為大于1，且小于或等于N的整數；基于每張白平衡標簽圖像和對應的混合色溫圖像，訓練預設深度學習模型，得到深度學習白平衡校正模型。

一種可能的實現方式中，獲取多張單色溫圖像；其中，每張單色溫圖像內附帶色卡；基于每張單色溫圖像內的色卡，確定每張單色溫圖像對應的光源估計值；基于每張單色溫圖像的光源估計值，對每張單色溫圖像進行白平衡校正，得到每張單色溫圖像對應的白平衡標簽圖像。