[發明專利]一種白平衡校正方法、裝置、設備和存儲介質

有效

• 申請號: 201910760080.3
• 申請日: 2019-08-16
• 公開（公告）號: CN112399162B
• 公開（公告）日: 2022-09-16
• 發明（設計）人: 范慶杰
• 申請（專利權）人: 浙江宇視科技有限公司
• 主分類號: H04N9/73
• 分類號: H04N9/73;G06T5/00
• 代理公司: 北京品源專利代理有限公司 11332
• 代理人: 孟金喆

本發明實施例公開了一種白平衡校正方法、裝置、設備和存儲介質。其中，方法包括：獲取預先訓練好的光源估計模型；針對任一攝像設備，基于所述光源估計模型構建自適應校正模型，并利用所述攝像設備采集的樣本圖像訓練所述自適應校正模型；基于訓練好的所述自適應校正模型對所述攝像設備采集的圖像進行白平衡校正。本發明實施例基于預先訓練好的光源估計模型構建適用于攝像設備的自適應校正模型，實現了在較小訓練樣本的情況下，使網絡方便地移植到不同的設備上，適應性強，并實現白平衡的準確校正。

技術領域

本發明實施例涉及圖像處理領域，尤其涉及一種白平衡校正方法、裝置、設備和存儲介質。

背景技術

由于人眼具有獨特的適應性，因此無論在室外陽光、室內熒光燈或者混合光源下，人們看到的白色的物體依然為白色。但是，攝像機沒有辦法像人眼自我修正光線的變化，因此需要做白平衡處理。

當前，應用較為廣泛的白平衡校正算法，是基于統計的方法。即在某種假設下，通過統計場景圖像自身的顏色特征來估計出環境光源。如灰度世界算法，該算法認為所有物理表面的平均反射是無色差的，也就是說，在RGB三個通道下，場景中的對三個顏色通道的平均反射率是相等的。完美反射法，該算法假設認為：一幅圖像中，顏色通道的最大響應是由場景中的白色表面引起的。由于白色表面能夠完全反映出場景光照的顏色，因此，三個顏色通道的最大值將被用作圖像的光照顏色。但基于統計的算法的方式仍存在一定的不足：普遍存在假設性較強，自適應能力差。

發明內容

本發明實施例提供一種白平衡校正方法、裝置、設備和存儲介質，以解決現有技術中實現白平衡校正的方法的自適應性差的技術問題。

第一方面，本發明實施例提供了一種白平衡校正方法，該方法包括：

獲取預先訓練好的光源估計模型；

針對任一攝像設備，基于所述光源估計模型構建自適應校正模型，并利用所述攝像設備采集的樣本圖像訓練所述自適應校正模型；

基于訓練好的所述自適應校正模型對所述攝像設備采集的圖像進行白平衡校正。

第二方面，本發明實施例還提供了一種白平衡校正裝置，該裝置包括：

獲取模塊，用于獲取預先訓練好的光源估計模型；

遷移模塊，用于針對任一攝像設備，基于所述光源估計模型構建自適應校正模型，并利用所述攝像設備采集的樣本圖像訓練所述自適應校正模型；

校正模塊，用于基于訓練好的所述自適應校正模型對所述攝像設備采集的圖像進行白平衡校正。第三方面，本發明實施例還提供了一種設備，包括：

一個或多個處理器；

存儲裝置，用于存儲一個或多個程序，

當所述一個或多個程序被所述一個或多個處理器執行，使得所述一個或多個處理器實現如本發明任一實施例所述的白平衡校正方法。

第四方面，本發明實施例還提供了一種計算機可讀存儲介質，其上存儲有計算機程序，該程序被處理器執行時實現如本發明任一實施例所述的白平衡校正方法。

本發明實施例基于預先訓練好的光源估計模型構建適用于攝像設備的自適應校正模型，實現了在較小訓練樣本的情況下，使的訓練好的自適應校正模型方便地移植到攝像設備上，而且通過訓練好的自適應校正模型可實現攝像設備白平衡的準確校正。

附圖說明

圖1a為本發明實施例一中的白平衡校正方法的流程圖；

圖1b是本發明實施例一中的光源估計模型的結構圖；

圖2是本發明實施例二中的白平衡校正方法的流程圖；

圖3a是本發明實施例三中的白平衡校正方法的流程圖；