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

在審

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

1.一種圖像白平衡校正方法，其特征在于，所述方法包括：

獲取待校準圖像；

基于N種預設環境光源對所述待校準圖像進行渲染，得到N張第一圖像；其中，第一圖像與預設環境光源一一對應，所述N為大于1的整數；

將所述N張第一圖像輸入深度學習白平衡校正模型，得到所述待校準圖像對應的白平衡校正圖像。

2.根據權利要求1所述方法，其特征在于，所述待校準圖像包括單色溫圖像或混合色溫圖像。

3.根據權利要求1或2所述方法，其特征在于，所述將所述N張第一圖像輸入深度學習白平衡校正模型，得到所述待校準圖像對應的白平衡校正圖像，包括：

將所述N張第一圖像輸入所述深度學習白平衡校正模型，得到N個第一權重圖像；其中，所述第一權重圖像與所述第一圖像一一對應；

基于所述N個第一權重圖像對所述N張第一圖像進行融合處理，得到所述待校準圖像對應的白平衡校正圖像。

4.根據權利要求1-3中任一項所述方法，其特征在于，在所述將所述N張第一圖像輸入深度學習白平衡校正模型，得到所述待校準圖像對應的白平衡校正圖像之前，所述方法還包括：

獲取多張單色溫圖像中每張單色溫圖像對應的白平衡標簽圖像；

基于M種預設環境光源對每張白平衡標簽圖像進行渲染，得到所述每張白平衡標簽圖像對應的混合色溫圖像；其中，所述M為大于1，且小于或等于N的整數；

基于所述每張白平衡標簽圖像和對應的混合色溫圖像，訓練預設深度學習模型，得到所述深度學習白平衡校正模型。

5.根據權利要求4所述方法，其特征在于，所述獲取多張單色溫圖像中每張單色溫圖像對應的白平衡標簽圖像，包括：

獲取多張單色溫圖像；其中，每張單色溫圖像內附帶色卡；

基于所述每張單色溫圖像內的色卡，確定所述每張單色溫圖像對應的光源估計值；

基于所述每張單色溫圖像的光源估計值，對所述每張單色溫圖像進行白平衡校正，得到所述每張單色溫圖像對應的白平衡標簽圖像。

6.根據權利要求4或5所述方法，其特征在于，所述基于M種預設環境光源對每張白平衡標簽圖像進行渲染，得到所述每張白平衡標簽圖像對應的混合色溫圖像，包括：

基于所述M種預設環境光源對所述每張白平衡標簽圖像進行渲染，得到所述每張白平衡標簽圖像對應的M張第二圖像；其中，所述第二圖像與所述預設環境光源一一對應；

獲取所述每張白平衡標簽圖像對應的M個第二權重圖像；

基于所述M個第二權重圖像對所述M張第二圖像進行融合處理，得到所述每張白平衡標簽圖像對應的混合色溫圖像。

7.根據權利要求6所述方法，其特征在于，所述基于所述每張白平衡標簽圖像和對應的混合色溫圖像，訓練預設深度學習模型，得到所述深度學習白平衡校正模型，包括：

基于所述N種預設環境光源對所述每張白平衡標簽圖像對應的混合色溫圖像進行渲染，得到N張第三圖像；其中，所述第三圖像與所述預設環境光源一一對應；

基于所述N張第三圖像和對應的白平衡標簽圖像，訓練預設深度學習模型，得到所述深度學習白平衡校正模型。

8.一種圖像白平衡校正裝置，其特征在于，所述圖像白平衡校正裝置包括：

圖像獲取單元，用于獲取待校準圖像；

圖像渲染單元，用于基于N種預設環境光源對所述待校準圖像進行渲染，得到N張第一圖像；其中，第一圖像與預設環境光源一一對應，所述N為大于1的整數；

白平衡校正單元，用于將所述N張第一圖像輸入深度學習白平衡校正模型，得到所述待校準圖像對應的白平衡校正圖像。

9.根據權利要求8所述裝置，其特征在于，所述待校準圖像包括單色溫圖像或混合色溫圖像。