[发明专利]SDR视频转换为HDR视频的方法、装置、设备及存储介质

权利要求书

1.一种SDR视频转换为HDR视频的方法，其特征在于，包括：

获取SDR视频，将SDR视频解码，得到若干一帧SDR图像；

构建自适应3D LUT映射表，将每帧SDR图像输入自适应3D LUT模型，采用三线性插值法将SDR图像转换为HDR图像，得到多个一帧HDR图像；

将多帧HDR图像进行编码，得到HDR视频；

其中，构建自适应3D LUT映射表进一步包括：

构建取值范围在[-1,1]的3D LUT映射表，所述映射表为三维结构，包含三个坐标分量，将所述三个坐标分量网格化，得到初步的视频转换模型；对视频转换模型进行训练，调整多元线性方程的回归参数，得到自适应3D LUT映射表；

所述对视频转换模型进行训练进一步包括：

通过公式：

对视频转换模型进行监督学习，不断计算图像gt与图像y的loss值，对多元线性方程的回归参数进行调整，实现输入SDR图像回归曲线的拟合；其中，N为输入SDR图像的像素个数，图像gt为模型训练集中HDR视频解码后的图像，图像y为视频转换模型输出的图像。

2.如权利要求1所述的SDR视频转换为HDR视频的方法，其特征在于，所述自适应3D LUT模型的算法公式为：

lut＝regressLUT(x)

y＝trilinear(x,lut)

其中，x为SDR视频解码后的RGB图像，lut为输入图像x的多元线性回归拟合，y为输入图像x通过三线性插值后得到的图像。

3.一种SDR视频转换为HDR视频的装置，实现如权利要求1或2所述的SDR视频转换为HDR视频的方法，其特征在于，包括：

SDR模块，用于获取SDR视频，将SDR视频解码，得到若干一帧SDR图像；

转换模块，用于构建自适应3D LUT映射表，将每帧SDR图像输入自适应3D LUT模型，采用三线性插值法将SDR图像转换为HDR图像，得到多个一帧HDR图像；

HDR模块，用于将多帧HDR图像进行编码，得到HDR视频。

4.如权利要求3所述的SDR视频转换为HDR视频的装置，其特征在于，所述转换模块包括映射表构建单元及模型训练单元；

所述映射表构建单元用于构建取值范围在[-1,1]的3D LUT映射表，所述映射表为三维结构，包含三个坐标分量，将所述三个坐标分量网格化，得到初步的视频转换模型；对视频转换模型进行训练，调整多元线性方程的回归参数，得到自适应3D LUT映射表；

所述模型训练单元用于根据公式：

对视频转换模型进行监督学习，不断计算图像gt与图像y的loss值，对多元线性方程的回归参数进行调整，实现输入SDR图像回归曲线的拟合；其中，N为输入SDR图像的像素个数，图像gt为模型训练集中HDR视频解码后的图像，图像y为视频转换模型输出的图像。

5.一种SDR视频转换为HDR视频的设备，其特征在于，包括：

存储器和处理器，所述存储器中存储有指令，所述存储器和所述处理器通过线路互连；

所述处理器调用所述存储器中的所述指令，实现如权利要求1或2所述的SDR视频转换为HDR视频的方法。

6.一种计算机可读存储介质，所述计算机可读存储介质上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1或2所述的SDR视频转换为HDR视频的方法。