[發明專利]瞳孔亮斑消除方法

有效

• 申請號: 202110674229.3
• 申請日: 2021-06-17
• 公開（公告）號: CN113409210B
• 公開（公告）日: 2023-06-02
• 發明（設計）人: 葉維健;汪丹丹;劉剛;曾峰
• 申請（專利權）人: 杭州海康威視數字技術股份有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/50
• 代理公司: 北京柏杉松知識產權代理事務所(普通合伙) 11413
• 代理人: 孫翠賢;孟維娜

本發明實施例提供了瞳孔亮斑消除方法，應用于圖像處理技術領域。該方法包括：獲取目標圖像；其中，目標圖像包含具有瞳孔亮斑的瞳孔區域；針對目標圖像中瞳孔區域的每一分區，利用預先訓練的瞳孔區域的分區與像素值之間的目標映射關系，將該分區內的像素點的像素值校正為該分區所映射的像素值；每一分區所映射的像素值為該分區的像素點在消除亮斑后的像素值。通過本方案，可以消除圖像中存在瞳孔亮斑，提升圖像的成像效果。

技術領域

本發明涉及圖像處理技術領域，特別是涉及瞳孔亮斑消除方法。

背景技術

隨著技術的發展，成像技術越來越廣泛的被應用在各行各業，如通過手機進行拍照、通過攝像頭進行視頻監控等。

一般來說，在光照條件較好(光源充足)的情況下，成像設備所生成的圖像的成像質量較好，而在光照條件較差(如黑夜)的情況下，成像設備所生成的圖像往往包含較多的噪音，成像質量較差。為了提高在光照條件較差的情況下圖像的成像質量，可以利用特殊的分光結構和雙光融合技術，使得所生成的圖像融合了可見光圖像的顏色信息和紅外圖像的信噪比優勢，進而提高圖像的成像質量。

然而，雙光融合技術需要成像設備在采集圖像時額外的進行補光，導致所生成的圖像中存在瞳孔亮斑(瞳孔處有明顯亮斑)的現象，影響圖像的成像效果。

發明內容

本發明實施例的目的在于提供瞳孔亮斑消除方法，以消除圖像中存在瞳孔亮斑，提升圖像的成像效果。具體技術方案如下：

第一方面，本發明實施例提供了一種瞳孔亮斑消除方法，所述方法包括：

獲取目標圖像；其中，所述目標圖像包含具有瞳孔亮斑的瞳孔區域；

針對所述目標圖像中瞳孔區域的每一分區，利用預先訓練的瞳孔區域的分區與像素值之間的目標映射關系，將該分區內的像素點的像素值校正為該分區所映射的像素值；每一分區所映射的像素值為該分區的像素點在消除亮斑后的像素值；

其中，所述目標映射關系的訓練方法包括：

獲取多張第一圖像和第一圖像對應的第二圖像；所述第一圖像中包含瞳孔亮斑，所述第二圖像為未包含瞳孔亮斑的圖像，且所述第一圖像和相應的第二圖像除瞳孔區域以外的圖像區域相同；

基于多張第一圖像和每一第一圖像對應的第二圖像，對預設的瞳孔區域的分區與像素值之間的映射關系進行訓練，得到所述目標映射關系；

其中，利用所述目標映射關系進行校正后的每一第一圖像的瞳孔區域內各像素點的像素值與像素值真值之間的損失函數值之和小于預設閾值，每一像素點的像素值真值為該第一圖像相對應的第二圖像瞳孔區域內，與該像素點具有相同位置的像素點的像素值，所述損失函數值基于像素值損失所確定。

可選的，在一實施例中，所述基于多張第一圖像和第一圖像對應的第二圖像，對預設的關于瞳孔區域內各分區與像素值的映射關系進行訓練，得到所述目標映射關系，包括：

針對每一第一圖像，基于如下方式對預設的關于瞳孔區域內各位置與像素值的映射關系進行訓練：

基于預設的關于瞳孔區域內各分區與像素值的映射關系，確定該第一圖像的瞳孔區域內各分區映射至的像素值，得到校正后的該第一圖像；

針對校正后的該第一圖像的瞳孔區域內每一像素點，計算該像素點的像素值，與該像素點的像素值真值的損失函數值，作為該像素點的損失函數值；

判斷校正后的該第一圖像的瞳孔區域內各像素點的損失函數值的和值是否小于預設閾值，當所述和值不小于所述預設閾值時，調整所述關于瞳孔區域內各分區與像素值的映射關系，并進行下一次訓練，當所述和值小于所述預設閾值時，得到訓練完成的目標映射關系。