[发明专利]一种HDR视频转换方法、装置、设备及计算机存储介质

说明书

本申请提供一种HDR视频转换方法、装置、设备及计算机存储介质。涉及视频处理技术领域，上述方法包括：对待处理的SDR视频进行抽帧处理，得到SDR视频中包含的M帧SDR视频帧；利用已训练的HDR视频转换模型将M帧SDR视频帧转换为M帧HDR视频帧，其中，M帧HDR视频帧中的第t帧HDR视频帧是根据SDR视频中包含第t帧SDR视频帧在内的K帧SDR视频帧转换得到，0t≤M，t、K、M均为正整数；将M帧HDR视频帧进行合帧处理，得到与SDR视频对应的HDR视频。本申请在SDR视频帧转换过程中尽可能保留了SDR视频帧之间的信息，避免了SDR视频帧转换过程中信息的损失，提高了HDR视频的转换质量。

技术领域

本申请属于视频处理技术领域，尤其涉及一种HDR视频转换方法、装置、设备及计算机存储介质。

背景技术

相比于标准动态范围(Standard Dynamic Range，SDR)视频，高动态范围(HighDynamic Range，HDR)视频具有更大的动态范围以及更宽广的色域，能够展现出更高的对比度以及更丰富的色彩。随着HDR视频的拍摄及显示技术的逐渐成熟，越来越多的播放设备支持HDR视频的播放。因此，如何将SDR视频转换为HDR视频，成为当前的热门问题。

由于深度学习方法在图像处理和计算机视觉领域的广泛应用，目前提出的HDR视频转换方法是，首先利用神经网络将SDR视频中的每一帧SDR视频帧转换为HDR视频帧，然后通过合帧的方式获取转换后的HDR视频。但通过这种方式得到的HDR视频的质量较差。

发明内容

本申请实施例提供一种HDR视频转换方法、装置、终端设备及存储介质，可以解决由SDR视频转换得到的HDR视频质量较低的问题。

第一方面，本申请实施例提供了一种HDR视频转换方法，该方法包括：对待处理的SDR视频进行抽帧处理，得到所述SDR视频中包含的M帧SDR视频帧；

利用已训练的HDR视频转换模型将M帧SDR视频帧转换为M帧HDR视频帧，其中，所述M帧HDR视频帧中的第t帧HDR视频帧是根据所述SDR视频中包含第t帧SDR视频帧在内的K帧SDR视频帧转换得到，0t≤M，t、K、M均为正整数；

将所述M帧HDR视频帧进行合帧处理，得到与所述SDR视频对应的HDR视频。

在本申请提供的HDR视频转换方法中，每一帧HDR视频帧均是由SDR视频中的K帧SDR视频帧转换得到的。可以理解的是，在视频中，由于光照的变化，帧与帧之间可能会存在一些帧间信息，使得前一帧损失的信息可能在后一帧中有所保留。因此，采用本申请提供的HDR视频转换方法，在转换HDR视频帧时，充分考虑了K帧SDR视频帧之间的信息，减小了SDR视频帧转换过程中的信息损失，使得处理后的HDR视频帧也尽可能保留了帧间信息，从而提高了HDR视频的转换质量。

可选地，所述K帧SDR视频帧是连续的SDR视频帧。

可选地，所述K帧SDR视频帧包括所述SDR视频中的第t-k帧SDR视频帧到第t+k帧SDR视频帧，其中，2k+1＝K。

可选地，若所述SDR视频中位于所述第t帧SDR视频帧之前的SDR视频帧数为n，0≤n＜k，则所述K帧SDR视频帧包括k-n+1帧所述SDR视频中的第1帧SDR视频帧。

可选地，若所述SDR视频中位于所述第t帧SDR视频帧之后的SDR视频帧数为p，0≤p＜k，则所述K帧SDR视频帧包括k-p+1帧所述SDR视频中的第M帧SDR视频帧。

第二方面，本申请实施例提供了一种HDR视频转换装置，该装置包括：抽帧单元，用于对待处理的SDR视频进行抽帧处理，得到所述SDR视频中包含的M帧SDR视频帧；