[發(fā)明專利]瞳孔亮斑消除方法

有效

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

1.一種瞳孔亮斑消除方法，其特征在于，所述方法包括：

獲取目標(biāo)圖像；其中，所述目標(biāo)圖像包含具有瞳孔亮斑的瞳孔區(qū)域；

針對所述目標(biāo)圖像中瞳孔區(qū)域的每一分區(qū)，利用預(yù)先訓(xùn)練的瞳孔區(qū)域的分區(qū)與像素值之間的目標(biāo)映射關(guān)系，將該分區(qū)內(nèi)的像素點的像素值校正為該分區(qū)所映射的像素值；每一分區(qū)所映射的像素值為該分區(qū)的像素點在消除亮斑后的像素值；

其中，所述目標(biāo)映射關(guān)系的訓(xùn)練方法包括：

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

基于多張第一圖像和每一第一圖像對應(yīng)的第二圖像，對預(yù)設(shè)的瞳孔區(qū)域的分區(qū)與像素值之間的映射關(guān)系進行訓(xùn)練，得到所述目標(biāo)映射關(guān)系；

其中，利用所述目標(biāo)映射關(guān)系進行校正后的每一第一圖像的瞳孔區(qū)域內(nèi)各像素點的像素值與像素值真值之間的損失函數(shù)值之和小于預(yù)設(shè)閾值，每一像素點的像素值真值為該第一圖像相對應(yīng)的第二圖像瞳孔區(qū)域內(nèi)，與該像素點具有相同位置的像素點的像素值，所述損失函數(shù)值基于像素值損失所確定；

所述獲取多張第一圖像和第一圖像對應(yīng)的第二圖像，包括：

通過以下方式獲取每一第一圖像和第一圖像對應(yīng)的第二圖像：

獲取第一初始圖像和所述第一初始圖像對應(yīng)的第二初始圖像；其中，所述第一初始圖像通過具有補光的第一攝像機采集得到，所述第二初始圖像通過無補光的第二攝像機采集得到，所述第一攝像機和所述第二攝像機在采集所述第一初始圖像和所述第二初始圖像時具有相同的配置參數(shù)、且所針對的對象相同；

基于所述第一初始圖像的瞳孔區(qū)域的區(qū)域信息，更新所述第二初始圖像中的瞳孔區(qū)域，得到更新后的圖像，作為第一圖像，并將所述第二初始圖像作為與該第一圖像對應(yīng)的第二圖像；其中，所述更新后的圖像的瞳孔區(qū)域與所述第一初始圖像的瞳孔區(qū)域相同，且所述更新后的圖像除瞳孔區(qū)域以外的圖像區(qū)域與所述第二初始圖像相同。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述基于多張第一圖像和第一圖像對應(yīng)的第二圖像，對預(yù)設(shè)的關(guān)于瞳孔區(qū)域內(nèi)各分區(qū)與像素值的映射關(guān)系進行訓(xùn)練，得到所述目標(biāo)映射關(guān)系，包括：

針對每一第一圖像，基于如下方式對預(yù)設(shè)的關(guān)于瞳孔區(qū)域內(nèi)各位置與像素值的映射關(guān)系進行訓(xùn)練：

基于預(yù)設(shè)的關(guān)于瞳孔區(qū)域內(nèi)各分區(qū)與像素值的映射關(guān)系，確定該第一圖像的瞳孔區(qū)域內(nèi)各分區(qū)映射至的像素值，得到校正后的該第一圖像；

針對校正后的該第一圖像的瞳孔區(qū)域內(nèi)每一像素點，計算該像素點的像素值，與該像素點的像素值真值的損失函數(shù)值，作為該像素點的損失函數(shù)值；

判斷校正后的該第一圖像的瞳孔區(qū)域內(nèi)各像素點的損失函數(shù)值的和值是否小于預(yù)設(shè)閾值，當(dāng)所述和值不小于所述預(yù)設(shè)閾值時，調(diào)整所述關(guān)于瞳孔區(qū)域內(nèi)各分區(qū)與像素值的映射關(guān)系，并進行下一次訓(xùn)練，當(dāng)所述和值小于所述預(yù)設(shè)閾值時，得到訓(xùn)練完成的目標(biāo)映射關(guān)系。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述針對校正后的該第一圖像的瞳孔區(qū)域內(nèi)每一像素點，計算該像素點的像素值，與該像素點的像素值真值的損失函數(shù)值，作為該像素點的損失函數(shù)值，包括：

針對校正后的該第一圖像的瞳孔區(qū)域內(nèi)每一像素點，計算該像素點像素值，與該像素點的像素值真值的像素值損失，并基于所確定的像素值損失，確定該像素點的像素值，與該像素點的像素值真值的損失函數(shù)值，作為該像素點的損失函數(shù)值。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述基于所確定的像素值損失，確定該像素點的像素值，與該像素點的像素值真值的損失函數(shù)值，包括：

將所確定的像素值損失，確定為該像素點的像素值，與該像素點的像素值真值的損失函數(shù)值；

或者，

計算該像素點的輔助損失；基于所確定的像素值損失和計算得到的輔助損失，確定該像素點的像素值，與該像素點的像素值真值的損失函數(shù)值；其中，所述輔助損失包括該像素點的顏色損失和平滑度損失中的至少一種。

5.根據(jù)權(quán)利要求4所述的方法，其特征在于，所述計算該像素點的輔助損失，包括：

計算該像素點的像素值與該像素點的像素值真值在色彩空間中的夾角，作為該像素點的顏色損失；和/或，

計算該像素點的梯度，作為該像素點的平滑度損失。