[发明专利]瞳孔亮斑消除方法

权利要求书

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

获取目标图像；其中，所述目标图像包含具有瞳孔亮斑的瞳孔区域；

针对所述目标图像中瞳孔区域的每一分区，利用预先训练的瞳孔区域的分区与像素值之间的目标映射关系，将该分区内的像素点的像素值校正为该分区所映射的像素值；每一分区所映射的像素值为该分区的像素点在消除亮斑后的像素值；

其中，所述目标映射关系的训练方法包括：

获取多张第一图像和第一图像对应的第二图像；所述第一图像中包含瞳孔亮斑，所述第二图像为未包含瞳孔亮斑的图像，且所述第一图像和相应的第二图像除瞳孔区域以外的图像区域相同；

基于多张第一图像和每一第一图像对应的第二图像，对预设的瞳孔区域的分区与像素值之间的映射关系进行训练，得到所述目标映射关系；

其中，利用所述目标映射关系进行校正后的每一第一图像的瞳孔区域内各像素点的像素值与像素值真值之间的损失函数值之和小于预设阈值，每一像素点的像素值真值为该第一图像相对应的第二图像瞳孔区域内，与该像素点具有相同位置的像素点的像素值，所述损失函数值基于像素值损失所确定；

所述获取多张第一图像和第一图像对应的第二图像，包括：

通过以下方式获取每一第一图像和第一图像对应的第二图像：

获取第一初始图像和所述第一初始图像对应的第二初始图像；其中，所述第一初始图像通过具有补光的第一摄像机采集得到，所述第二初始图像通过无补光的第二摄像机采集得到，所述第一摄像机和所述第二摄像机在采集所述第一初始图像和所述第二初始图像时具有相同的配置参数、且所针对的对象相同；

基于所述第一初始图像的瞳孔区域的区域信息，更新所述第二初始图像中的瞳孔区域，得到更新后的图像，作为第一图像，并将所述第二初始图像作为与该第一图像对应的第二图像；其中，所述更新后的图像的瞳孔区域与所述第一初始图像的瞳孔区域相同，且所述更新后的图像除瞳孔区域以外的图像区域与所述第二初始图像相同。

2.根据权利要求1所述的方法，其特征在于，所述基于多张第一图像和第一图像对应的第二图像，对预设的关于瞳孔区域内各分区与像素值的映射关系进行训练，得到所述目标映射关系，包括：

针对每一第一图像，基于如下方式对预设的关于瞳孔区域内各位置与像素值的映射关系进行训练：

基于预设的关于瞳孔区域内各分区与像素值的映射关系，确定该第一图像的瞳孔区域内各分区映射至的像素值，得到校正后的该第一图像；

针对校正后的该第一图像的瞳孔区域内每一像素点，计算该像素点的像素值，与该像素点的像素值真值的损失函数值，作为该像素点的损失函数值；

判断校正后的该第一图像的瞳孔区域内各像素点的损失函数值的和值是否小于预设阈值，当所述和值不小于所述预设阈值时，调整所述关于瞳孔区域内各分区与像素值的映射关系，并进行下一次训练，当所述和值小于所述预设阈值时，得到训练完成的目标映射关系。

3.根据权利要求2所述的方法，其特征在于，所述针对校正后的该第一图像的瞳孔区域内每一像素点，计算该像素点的像素值，与该像素点的像素值真值的损失函数值，作为该像素点的损失函数值，包括：

针对校正后的该第一图像的瞳孔区域内每一像素点，计算该像素点像素值，与该像素点的像素值真值的像素值损失，并基于所确定的像素值损失，确定该像素点的像素值，与该像素点的像素值真值的损失函数值，作为该像素点的损失函数值。

4.根据权利要求3所述的方法，其特征在于，所述基于所确定的像素值损失，确定该像素点的像素值，与该像素点的像素值真值的损失函数值，包括：

将所确定的像素值损失，确定为该像素点的像素值，与该像素点的像素值真值的损失函数值；

或者，

计算该像素点的辅助损失；基于所确定的像素值损失和计算得到的辅助损失，确定该像素点的像素值，与该像素点的像素值真值的损失函数值；其中，所述辅助损失包括该像素点的颜色损失和平滑度损失中的至少一种。

5.根据权利要求4所述的方法，其特征在于，所述计算该像素点的辅助损失，包括：

计算该像素点的像素值与该像素点的像素值真值在色彩空间中的夹角，作为该像素点的颜色损失；和/或，

计算该像素点的梯度，作为该像素点的平滑度损失。