[發明專利]光源估計方法、裝置、電子設備、芯片及存儲介質

在審

• 申請號: 202210593898.2
• 申請日: 2022-05-27
• 公開（公告）號: CN116385566A
• 公開（公告）日: 2023-07-04
• 發明（設計）人: 張雪巖
• 申請（專利權）人: 上海玄戒技術有限公司
• 主分類號: G06T7/90
• 分類號: G06T7/90;G06T5/00
• 代理公司: 北京法勝知識產權代理有限公司 11922
• 代理人: 戎鄭華

本公開涉及計算機視覺技術領域，具體提供了一種光源估計方法、裝置、電子設備、芯片、及存儲介質。一種光源估計方法，包括：獲取當前場景光源的多光譜數據；根據預先設置的標準光源數據集和多光譜數據，確定標準光源數據集對應的第一權重集；其中，標準光源數據集包括至少兩個標準光源的光譜數據，第一權重集包括每個標準光源對應的第一權值；根據標準光源數據集中各標準光源的光譜數據和第一權值，確定當前場景光源的估計結果。本公開實施方式中，提高光源估計準確性，圖像色彩還原與人眼實際感知到的色彩更為接近，提高成像效果。

技術領域

本公開涉及圖像處理技術領域，具體涉及一種光源估計方法、裝置、電子設備、芯片及存儲介質。

背景技術

對于人體視覺系統，在場景光源變化的情況下，人眼能夠不受場景光源變化的影響而感知到物體自身固有的顏色，這種能力被稱為顏色恒常性。而對于計算機視覺系統，在場景光源發生變化的時候，圖像傳感器的成像效果會出現不同程度的色偏，例如物體偏藍、偏黃等。

光源估計算法可以對當前場景光源的相關色溫或白點坐標進行估計，從而將場景圖像恢復至標準光源下，實現圖像的色彩還原。但是，相關技術中，光源估計效果較差，導致最終還原的圖像色彩與人眼實際感知存在偏差，圖像色彩效果不佳。

發明內容

為提高光源估計準確性，本公開實施方式提供了一種光源估計方法、裝置、電子設備、芯片及存儲介質。

第一方面，本公開實施方式提供了一種光源估計方法，包括：

獲取當前場景光源的多光譜數據；

根據預先設置的標準光源數據集和所述多光譜數據，確定所述標準光源數據集對應的第一權重集；其中，所述標準光源數據集包括至少兩個標準光源的光譜數據，所述第一權重集包括每個標準光源對應的第一權值；

根據所述標準光源數據集中各標準光源的所述光譜數據和所述第一權值，確定所述當前場景光源的估計結果。

在一些實施方式中，所述根據預先設置的標準光源數據集和所述多光譜數據，確定所述標準光源數據集對應的第一權重集，包括：

根據所述標準光源數據集和所述多光譜數據，對每個標準光源的初始權值進行回歸處理，得到每個標準光源對應的第一權值；

根據每個標準光源對應的第一權值，確定所述第一權重集。

在一些實施方式中，所述標準光源數據集包括每個標準光源的相關色溫數據；所述根據所述標準光源數據集中各標準光源的所述光譜數據和所述第一權值，確定所述當前場景光源的估計結果，包括：

根據所述第一權重集中的各個的所述第一權值，從所述標準光源數據集中確定至少兩個候選標準光源；

根據每個候選標準光源的所述光譜數據和所述相關色溫數據、以及所述多光譜數據，確定當前場景光源的估計結果。

在一些實施方式中，所述根據各個所述標準光源的所述第一權值，從所述標準光源數據集中確定至少兩個候選標準光源，包括：

對所述第一權重集中各個所述第一權值由大至小進行排序，得到第一排序集合；

根據所述第一排序集合中的最大值與預設權重閾值的關系，將排序前預設數量的第一權值對應的標準光源確定為所述候選標準光源。

在一些實施方式中，所述根據所述第一排序集合中的最大值與預設權重閾值的關系，將排序前預設數量的第一權值對應的標準光源確定為所述候選標準光源，包括：

響應于所述第一排序集合中的最大值不小于所述預設權重閾值，將排序前第一數量的第一權值對應的標準光源確定為所述候選標準光源；

和/或，