[發明專利]圖像白平衡校正方法、裝置、存儲介質及終端

有效

• 申請號: 201810934706.3
• 申請日: 2018-08-16
• 公開（公告）號: CN109040729B
• 公開（公告）日: 2020-04-07
• 發明（設計）人: 張弓
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: H04N9/64
• 分類號: H04N9/64;H04N9/73
• 代理公司: 北京品源專利代理有限公司 11332
• 代理人: 孟金喆

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

對目標圖像進行分割，得到多個目標子圖像；

獲取每個目標子圖像內容對應的光源方向；

根據所述多個目標子圖像對應光源方向判斷是否存在混光場景；

當存在混光場景時，根據所述混光場景對應的光源進行白平衡校正；

當不存在混光場景時，根據調白板進行白平衡處理；

所述獲取每個目標子圖像內容對應的光源方向，包括：

分別將每個目標子圖像輸入至預設機器學習模型，得到所述每個目標子圖像對應的光源方向。

2.根據權利要求1所述的圖像白平衡校正方法，其特征在于，所述根據所述多個目標子圖像對應光源方向判斷是否存在混光場景，包括：

判斷多個相鄰的目標子圖像對應的光源方向是否相同；

如果多個相鄰的目標子圖像對應的光源方向不相同，則存在混光場景。

3.根據權利要求1所述的圖像白平衡校正方法，其特征在于，所述根據所述混光場景對應的光源進行白平衡校正，包括：

根據所述多個目標子圖像對應的多個光源確定主光源；

根據所述主光源對所述目標圖像進行白平衡校正。

4.根據權利要求3所述的圖像白平衡校正方法，其特征在于，所述根據所述多個目標子圖像對應的多個光源確定主光源，包括：

獲取所述多個目標子圖像對應的多個光源方向；

統計每個光源方向對應的目標子圖像數量；

根據統計的目標子圖像數量確定主光源。

5.根據權利要求1所述的圖像白平衡校正方法，其特征在于，在分別將每個目標子圖像輸入值預設機器學習模型之前，還包括：

獲取多張具有單一光源的學習圖像；

將所述學習圖像輸入至卷積神經網絡模型中，得到預設機器學習模型。

6.根據權利要求1所述的圖像白平衡校正方法，其特征在于，所述對目標圖像進行分割，得到多個目標子圖像，包括：

獲取所述預設機器學習模型的輸入規格；

根據所述輸入規格對所述目標圖像進行分割，得到多個目標子圖像。

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

分割模塊，用于對目標圖像進行分割，得到多個目標子圖像；

獲取模塊，用于獲取所述分割模塊得到的每個目標子圖像內容對應的光源方向，獲取所述每個目標子圖像內容對應的光源方向，包括：

分別將每個目標子圖像輸入至預設機器學習模型，得到所述每個目標子圖像對應的光源方向；

判斷模塊，用于根據所述獲取模塊獲取的所述多個目標子圖像對應光源方向判斷是否存在混光場景；

白平衡模塊，用于當所述判斷模塊判定存在混光場景時，根據所述混光場景對應的光源進行白平衡校正；當不存在混光場景時，根據調白板進行白平衡處理。

8.一種計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，該程序被處理器執行時實現如權利要求1-6中任一所述圖像白平衡校正方法。

9.一種終端，包括存儲器，處理器及存儲在存儲器上并可在處理器運行的計算機程序，其特征在于，所述處理器執行所述計算機程序時實現如權利要求1-6任一所述的圖像白平衡校正方法。