[發明專利]圖像的白平衡處理方法、裝置、電子設備和存儲介質

有效

• 申請號: 202110427615.2
• 申請日: 2021-04-21
• 公開（公告）號: CN112866667B
• 公開（公告）日: 2021-07-23
• 發明（設計）人: 楊弢;劉闖
• 申請（專利權）人: 貝殼找房（北京）科技有限公司
• 主分類號: H04N9/73
• 分類號: H04N9/73
• 代理公司: 北京思源智匯知識產權代理有限公司 11657
• 代理人: 毛麗琴

本公開實施例公開了一種圖像的白平衡處理方法、裝置、電子設備和存儲介質。該圖像的白平衡處理方法包括：獲取待處理圖像；對上述待處理圖像進行圖像劃分，得到圖像區域集合；針對上述圖像區域集合中的圖像區域，通過對該圖像區域進行光源估計，得到該圖像區域的第一光源信息；確定該圖像區域的第一光源信息和目標光源信息之間的光源相似度，得到與該圖像區域相對應的光源相似度；基于與該圖像區域相對應的光源相似度，重新確定該圖像區域的光源信息，得到該圖像區域的第二光源信息；基于所得到的第二光源信息，對上述待處理圖像進行白平衡處理，生成白平衡處理后圖像。本公開實施例可以提高白平衡處理的準確度。

技術領域

本公開涉及圖像處理技術，尤其是一種圖像的白平衡處理方法、裝置、電子設備和存儲介質。

背景技術

白平衡（White Balance，WB），是描述圖像中白色精確度的一項指標。白平衡是隨著電子影像再現色彩而產生的。

實踐中，許多用戶在使用圖像拍攝裝置拍攝圖像的過程中，都會遇到這樣的問題：在日光燈的房間里拍攝的影像會顯得發綠，在室內鎢絲燈光下拍攝出來的景物就會偏黃，而在日光陰影處拍攝到的照片則莫名其妙地偏藍，其原因就在于白平衡算法的處理方式上。

目前，由于室內環境以其豐富的色彩表現，復雜的光源環境，大面積單色的場景等等給白平衡帶來了很大的挑戰。而且，在ISP（(Image Signal Processing，圖像信號處理）處理過程中，各個環節呈不同程度的耦合效應，導致在復雜的室內環境拍攝條件下，這種相互影響，相互耦合的效應被放大。例如，在多個光源共同作用下拍攝的圖像的白平衡處理難度較大，影響著白平衡處理的準確性。因此，在上述場景中，如何提高圖像白平衡處理的準確性，是一個值得關注的技術問題。

發明內容

本公開實施例提供一種圖像的白平衡處理方法、裝置、電子設備和存儲介質，以提高圖像白平衡處理的準確性。

根據本公開實施例的第一個方面，提供的一種圖像的白平衡處理方法，包括：

獲取待處理圖像；

對所述待處理圖像進行圖像劃分，得到圖像區域集合；

針對所述圖像區域集合中的圖像區域，通過對該圖像區域進行光源估計，得到該圖像區域的第一光源信息；確定該圖像區域的第一光源信息和目標光源信息之間的光源相似度，得到與該圖像區域相對應的光源相似度；其中，目標光源信息包括：所述圖像區域集合中除該圖像區域之外的其他圖像區域的第一光源信息，和/或，該圖像區域除第一光源信息之外的其他光源信息；基于與該圖像區域相對應的光源相似度，重新確定該圖像區域的光源信息，得到該圖像區域的第二光源信息；

基于所得到的第二光源信息，對所述待處理圖像進行白平衡處理，生成白平衡處理后圖像。

可選地，在本公開任一實施例的方法中，目標光源信息包括該圖像區域的偏色限制光源信息，偏色限制光源信息為用以限制偏色的光源信息；以及

所述確定該圖像區域的第一光源信息和目標光源信息之間的光源相似度，包括：

確定該圖像區域的第一光源信息和該圖像區域的偏色限制光源信息之間的光源相似度。

可選地，在本公開任一實施例的方法中，所述圖像區域集合中的圖像區域的色彩空間為第一色彩空間；以及

所述方法還包括：

基于該圖像區域的第一光源信息，確定與該圖像區域相對應的第一轉換矩陣，其中，與該圖像區域相對應的第一轉換矩陣用于將該圖像區域轉換為該圖像區域的白平衡處理后圖像區域；

將該圖像區域的白平衡處理后圖像區域的色彩空間轉換為第二色彩空間，得到該圖像區域的轉換后圖像區域；

對該圖像區域的轉換后圖像區域進行光源估計，得到該圖像區域的第三光源信息；