[发明专利]视频编解码中的环路滤波方法、装置、设备及存储介质

说明书

本申请提供了一种视频编解码中的环路滤波方法、装置、设备及存储介质，涉及视频编解码技术领域。所述方法包括：获取视频序列中的当前帧的类型；若当前帧的类型为帧内预测帧，则采用第一滤波模型对当前帧的重建帧进行滤波处理，得到滤波后的重建帧；其中，第一滤波模型是用于对帧内预测帧进行环路滤波处理的机器学习模型；若当前帧的类型为帧间预测帧，则采用第二滤波模型对当前帧的重建帧进行滤波处理，得到滤波后的重建帧；其中，第二滤波模型是用于对帧间预测帧进行环路滤波处理的机器学习模型。本申请实施例通过为I帧和非I帧分别选用相应的环路滤波模型，从而有助于提升滤波质量，特别是提升非I帧的滤波质量。

技术领域

本申请实施例涉及视频编解码技术领域，特别涉及一种视频编解码中的环路滤波方法、装置、设备及存储介质。

背景技术

在目前的视频编解码技术中，通过采用经离线训练的环路滤波模型对视频序列中的重建帧进行滤波处理，这种方法相较于传统滤波方法的滤波质量更优。

在相关技术中，通过构建环路滤波模型的训练集，采用该训练集对环路滤波模型进行离线训练。完成训练的环路滤波模型将被用于在视频序列的编解码过程中，对该视频序列中的重建帧进行滤波处理。

由于视频序列中的图像帧包括I帧和非I帧，非I帧包括P帧和B帧，在相关技术中，采用同一个环路滤波模型对I帧和非I帧进行滤波处理，导致滤波质量提升有限。

发明内容

本申请实施例提供了一种视频编解码中的环路滤波方法、装置、设备及存储介质，可以提升环路滤波模型的滤波质量。所述技术方案如下：

一方面，本申请实施例提供了一种视频编解码中的环路滤波方法，所述方法包括：

获取视频序列中的当前帧的类型，所述类型包括帧内预测帧和帧间预测帧；

若所述当前帧的类型为所述帧内预测帧，则采用第一滤波模型对所述当前帧的重建帧进行滤波处理，得到滤波后的重建帧；其中，所述第一滤波模型是用于对所述帧内预测帧进行环路滤波处理的机器学习模型；

若所述当前帧的类型为所述帧间预测帧，则采用第二滤波模型对所述当前帧的重建帧进行滤波处理，得到滤波后的重建帧；其中，所述第二滤波模型是用于对所述帧间预测帧进行环路滤波处理的机器学习模型。

另一方面，本申请实施例提供了一种视频编解码中的环路滤波模型的训练方法，所述方法包括：

生成第一训练集，所述第一训练集包括第一训练样本，所述第一训练样本的样本数据包括第一图像经帧内预测模式进行压缩重建后得到的第一重建图像，所述第一训练样本的标签数据包括所述第一图像；

采用所述第一训练集训练生成第一滤波模型，所述第一滤波模型是用于对视频序列中的帧内预测帧进行环路滤波处理的机器学习模型；

生成第二训练集，所述第二训练集包括第二训练样本，所述第二训练样本的样本数据包括第二图像经帧间预测模式进行压缩重建后得到的第二重建图像，所述第二训练样本的标签数据包括所述第二图像；

采用所述第二训练集训练生成第二滤波模型，所述第二滤波模型是用于对所述视频序列中的帧间预测帧进行环路滤波处理的机器学习模型。

另一方面，本申请实施例提供了一种视频编解码中的环路滤波装置，所述装置包括：

类型获取模块，用于获取视频序列中的当前帧的类型，所述类型包括帧内预测帧和帧间预测帧；

第一滤波模块，用于若所述当前帧的类型为所述帧内预测帧，则采用第一滤波模型对所述当前帧的重建帧进行滤波处理，得到滤波后的重建帧；其中，所述第一滤波模型是用于对所述帧内预测帧进行环路滤波处理的机器学习模型；