[发明专利]图像色调映射方法、装置、终端设备及存储介质

权利要求书

1.一种图像色调映射方法，其特征在于，所述图像色调映射方法包括：

获取高动态范围图像；

通过第一局部色调映射算法对所述高动态范围图像进行色调映射，得到第一低动态范围图像；

若所述高动态范围图像中存在目标区域，则从所述高动态范围图像中获取所述目标区域；所述目标区域是指所述高动态范围图像中通过所述第一局部色调映射算法进行色调映射后，存在光晕现象的区域；所述高动态范围图像中存在目标区域表征所述第一低动态范围图像中存在光晕现象；

通过已训练的色调映射模型对所述目标区域进行色调映射，得到第二低动态范围图像，所述第二低动态范围图像中不存在光晕现象；

将所述第一低动态范围图像和所述第二低动态范围图像进行融合，获取目标低动态范围图像；所述目标低动态范围图像是指所述第一低动态范围图像消除光晕现象后得到的图像。

2.如权利要求1所述的图像色调映射方法，其特征在于，在所述若所述高动态范围图像中存在目标区域，则从所述高动态范围图像中获取所述目标区域之前，包括：

检测所述高动态范围图像中是否存在太阳区域；

若所述高动态范围图像中存在太阳区域，则获取所述太阳区域的平均亮度值；

若所述太阳区域的平均亮度值大于亮度阈值，则确定所述高动态范围图像中存在所述目标区域，所述目标区域是根据所述太阳区域确定的。

3.如权利要求2所述的图像色调映射方法，其特征在于，所述图像色调映射方法还包括：

若所述高动态范围图像中不存在太阳区域，或者所述高动态范围图像中存在太阳区域且所述太阳区域的平均亮度值小于或等于亮度阈值，则检测所述第一低动态范围图像中是否存在光晕现象；

若所述第一低动态范围图像中存在光晕现象，则确定所述高动态范围图像中存在所述目标区域，所述目标区域是根据所述第一低动态范围图像中存在光晕现象的区域确定的。

4.如权利要求1所述的图像色调映射方法，其特征在于，所述色调映射模型通过以下方式进行训练：

获取训练样本；所述训练样本包括高动态范围样本图像和所述高动态范围样本图像对应的低动态范围样本图像；所述高动态范围样本图像是指通过所述第一局部色调映射算法进行色调映射后，所得的低动态范围图像中存在光晕现象的高动态范围图像；所述高动态范围样本图像对应的低动态范围样本图像是指所述高动态范围样本图像对应的不存在光晕现象的低动态范围图像；

将所述高动态范围样本图像作为待训练色调映射模型的输入图像，所述高动态范围样本图像对应的低动态范围样本图像作为所述待训练色调映射模型的真值图像，对所述待训练色调映射模型进行训练，得到所述已训练的色调映射模型。

5.如权利要求4所述的图像色调映射方法，其特征在于，所述获取训练样本包括：

获取至少两个待选择高动态范围图像；

通过所述第一局部色调映射算法，分别对所述至少两个待选择高动态范围图像进行色调映射，得到所述至少两个待选择高动态范围图像各自对应的低动态范围图像；

从所述至少两个待选择高动态范围图像中，获取对应的低动态范围图像中存在光晕现象的待选择高动态范围图像，并确定该待选择高动态范围图像为所述高动态范围样本图像；

对所述高动态范围样本图像对应的低动态范围图像进行预处理，消除所述高动态范围样本图像对应的低动态范围图像中的光晕现象，并确定已消除光晕现象的低动态范围图像为所述高动态范围样本图像对应的低动态范围样本图像；

或者通过至少一种第二局部色调映射算法对所述高动态范围样本图像进行色调映射，得到所述高动态范围样本图像对应的低动态范围样本图像，所述第二局部色调映射算法与所述第一局部色调映射算法不同，且所述高动态范围样本图像通过所述第二局部色调映射算法进行色调映射后，所得的低动态范围图像中不存在光晕现象。