一种能够去除带状伪像的视频逆色调映射方法及相关设备技术

本发明专利技术公开了一种能够去除带状伪像的视频逆色调映射方法及相关设备，方法包括：将SDR图像输入至第一神经网络，获取初步HDR亮度通道图；将SDR图像输入至语义分割模型，获取语义分割特征图，对初步HDR亮度通道图进行滤波，得到模糊图，根据模糊图获取对比特征图，根据初步HDR亮度通道图各个像素点的梯度获取带状伪像轨迹图；将初步HDR亮度通道图、语义分割特征图、对比特征图和带状伪像轨迹图进行组合，获取目标特征图；将目标特征图输入至掩膜生成网络，获取图像掩膜；根据图像掩膜融合初步HDR亮度通道图和模糊图，得到最终HDR亮度通道图；根据SDR图像和最终HDR亮度通道图获取目标HDR彩色图像。本发明专利技术可以消除带状伪像。本发明专利技术可以消除带状伪像。本发明专利技术可以消除带状伪像。

【技术实现步骤摘要】
一种能够去除带状伪像的视频逆色调映射方法及相关设备


[0001]本专利技术涉及图像处理
，特别涉及一种能够去除带状伪像的视频逆色调映射方法及相关设备。

技术介绍

[0002]高动态范围技术(HDR)发展迅速，相比标准动态范围技术(SDR)，HDR能够带来显著的视觉效果提升，现阶段HDR显示技术面临的发展障碍在于HDR内容非常缺乏，使得现有的HDR显示设备例如HDR电视等不能充分发挥作用，针对该问题，目前是采用逆色调映射技术(ITM)将现有的大量SDR图像转换为HDR图像，但目前的逆色调映射技术中，在将SDR图像转换为HDR图像时，在一些场景中会出现亮度的严重断层现象，即带状伪像。带状伪像的存在严重降低了转换后图像的的视觉效果。
[0003]因此，现有技术还有待改进和提高。

技术实现思路

[0004]针对现有技术的上述缺陷，本专利技术提供一种能够去除带状伪像的视频逆色调映射方法及相关设备，旨在解决现有技术中ITM技术中将SDR转换后得到的HDR图像中存在带状伪像的问题。
[0005]为了解决上述技术问题，本专本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种能够去除带状伪像的视频逆色调映射方法，其特征在于，所述方法包括：获取SDR图像，将所述SDR图像输入至第一神经网络，获取所述第一神经网络输出的初步HDR亮度通道图；将所述SDR图像输入至语义分割模型，获取所述语义分割模型输出的语义分割特征图，对所述初步HDR亮度通道图进行滤波，得到模糊图，根据所述模糊图和所述初步HDR亮度通道图的差值获取对比特征图，根据所述初步HDR亮度通道图各个像素点的梯度获取带状伪像轨迹图；将所述初步HDR亮度通道图、所述语义分割特征图、所述对比特征图和所述带状伪像轨迹图进行组合，获取目标特征图；将所述目标特征图输入至掩膜生成网络，获取所述掩膜生成网络输出的图像掩膜，所述图像掩膜中各个像素点的像素值反映像素点位于带状伪像区域的概率；根据所述图像掩膜融合所述初步HDR亮度通道图和所述模糊图，得到最终HDR亮度通道图；根据所述SDR图像和所述最终HDR亮度通道图获取目标HDR彩色图像。2.根据权利要求1所述的能够去除带状伪像的视频逆色调映射方法，其特征在于，所述根据所述初步HDR亮度通道图各个像素点的梯度获取带状伪像轨迹图，包括：根据所述初步HDR亮度通道图上每个像素点的相邻像素点获取各个像素点的初始梯度；根据所述初步HDR亮度通道图上每个像素点的初始梯度获取所述初步HDR亮度通道图上每个像素点的梯度的主方向强度；根据所述初始亮度通道图上每个像素点对应的所述初始梯度和所述主方向强度获取所述带状伪像轨迹图上每个像素点的像素值。3.根据权利要求2所述的能够去除带状伪像的视频逆色调映射方法，其特征在于，所述根据所述根据所述初步HDR亮度通道图上每个像素点的初始梯度获取所述初步HDR亮度通道图像每个像素点的梯度的主方向强度，包括：当所述初步HDR亮度通道图上的目标像素点在目标方向的所述初始梯度在预设范围内时，确定所述目标像素点在所述目标方向上的中间梯度为所述初始梯度；当所述初步HDR亮度通道图上的目标像素点在目标方向的梯度不在所述预设范围内时，确定所述目标像素点在所述目标方向上的中间梯度为0；确定所述目标像素点在各个目标方向上的中间梯度，将其中最大的所述中间梯度作为所述目标像素点的最终梯度；根据所述初步HDR亮度通道图上各个像素点的所述最终梯度确定各个像素点的所述主方向强度。4.根据权利要求3所述的能够去除带状伪像的视频逆色调映射方法，其特征在于，所述根据所述初步HDR亮度通道图上各个像素点的所述最终梯度确定各个像素点的所述主方向强度，包括：根据所述初步HDR亮度通道图上的各个像素点的所述初始梯度生成梯度图；对各个像素点所在的梯度图块进行变形处理，得到各个像素点的变形矢量，根据每个像素点的变形矢量构建目标矩阵；
根据各个像素点对应的所述目标矩阵的特征值和所述最终梯度得到各个像素点对应的所述主方向强度。5.根据权利要求1所述的能够去除带状伪像的视频逆色调映射方法，其特征在于，所述根据所述图像掩膜融合所述初步HDR亮度通道图和所述模糊图，得到最终HDR亮度通道图，包括：根据预设公式获取所述最终HDR亮度通道图；所述预设公式为：其中，Y
HDR
表示所述最终HDR亮度通道图，M为所述图像掩膜，Y
M
为所述模糊图，Y
D
为所述初步HDR亮度通道图。6.根据权利要求1所述的能够去除带状伪像的视频逆色调映射方法，其特征在于，所述根据所述SDR图像和所述最终HDR亮度通道图获取目标HDR彩色图像，包括：根据所述SDR图像和所述最终HDR亮度通道图获取目标亮度增益图；将所述SDR图像转...

【专利技术属性】
技术研发人员：梁志坚，周飞，袁树鸿，邱国平，
申请(专利权)人：深圳大学，
类型：发明
国别省市：