[發明專利]一種自動白平衡優化方法、設備以及計算機可讀存儲介質

有效

• 申請號: 202110210020.1
• 申請日: 2021-02-24
• 公開（公告）號: CN113099202B
• 公開（公告）日: 2023-05-12
• 發明（設計）人: 孫亮;王建淼;朱飛月
• 申請（專利權）人: 浙江大華技術股份有限公司
• 主分類號: H04N23/88
• 分類號: H04N23/88;H04N9/73
• 代理公司: 深圳市威世博知識產權代理事務所(普通合伙) 44280
• 代理人: 黎堅怡

本申請公開了一種自動白平衡優化方法、設備以及計算機可讀存儲介質，該自動白平衡優化方法包括：獲取圖像的YUV顏色空間的第一編碼數據；將YUV顏色空間的第一編碼數據轉換為預設顏色空間的第二編碼數據；按照預設尺寸將圖像劃分為多個圖像分塊；利用所述第二編碼數據對所述多個圖像分塊進行顏色識別，標記出預設顏色的圖像分塊；將標記出的圖像分塊的坐標數據以及圖像進行自動白平衡處理。通過上述方法，本申請可以通過將圖像轉換到魯棒性較好的其他顏色空間，然后對圖像進行顏色識別，利用顏色識別結果提高自動白平衡處理的準確性。

技術領域

本申請涉及圖像處理技術領域，特別是涉及一種自動白平衡優化方法、設備以及計算機可讀存儲介質。

背景技術

由于人眼具有獨特的適應性，在不同的色溫及光照下，人們觀察到的世界顏色計劃是一樣的。但是，對于視頻監控的攝像機，由于攝像機中的sensor傳感器、鏡頭等因素不能夠在某些環境變化下保持相對穩定的特性，所以攝像機無法在不同色溫下均能夠將圖像的顏色作準確的判斷，導致無法真實地監控客觀事物。自動白平衡技術就是將攝像機在低色溫下拍攝到的圖像整體顏色偏黃、在高色溫下拍攝到的圖像整體顏色偏藍的色偏進行校正。

目前主流的自動白平衡算法為基于先驗知識的自動白平衡算法，基于先驗知識的自動白平衡算法主要通過預先設置一個白區，將落入該區的統計點記作白點，那么沒有落入該白區的統計點記作非白點。這種方式通過色域映射法則可以將顏色進行校正，該算法色偏校正精度高，適用范圍廣。

但是，監控攝像機在實際的應用場景，例如，室外監控區域存在大面積綠色時，基于先驗知識的自動白平衡算法會將綠色統計點誤作為白點，導致整體圖像顏色偏色。

發明內容

本申請提供一種自動白平衡優化方法、設備以及計算機可讀存儲介質。

本申請提供的一個技術方案是：提供一種自動白平衡優化方法，所述自動白平衡優化方法包括：

獲取圖像的YUV顏色空間的第一編碼數據；

將YUV顏色空間的第一編碼數據轉換為預設顏色空間的第二編碼數據；

按照預設尺寸將所述圖像劃分為多個圖像分塊；

利用所述第二編碼數據對所述多個圖像分塊進行顏色識別，標記出預設顏色的圖像分塊；

將標記出的圖像分塊的坐標數據以及圖像進行自動白平衡處理。

在一些可能的實施例中，所述利用所述第二編碼數據對所述多個圖像分塊進行顏色識別，標記出預設顏色的圖像分塊的步驟，包括：

獲取所述第二編碼數據的至少一個通道數據；

將所述至少一個通道數據輸入預先訓練的分類器，其中，所述分類器事先對所述預設顏色進行訓練；

根據所述分類器的輸出結果獲取符合所述預設顏色的圖像分塊的坐標數據。

在一些可能的實施例中，所述自動白平衡優化方法，還包括：

提取所述圖像中符合所述預設顏色的特征數據；

利用所述特征數據訓練所述分類器；

其中，所述分類器為Weighted-KNN分類器。

在一些可能的實施例中，所述預設顏色空間為HSV顏色空間；

所述將所述至少一個通道數據輸入預先訓練的分類器的步驟，包括：

提取所述HSV顏色空間的第二編碼數據的H通道數據；

計算每個圖像分塊的H通道數據的眾數；

將所述每個圖像分塊對應的眾數輸入所述預先訓練的分類器；