[发明专利]一种视频画质增强的方法、装置、设备以及存储介质

说明书

本说明书提供一种视频画质增强的方法、装置、设备以及存储介质，该方法中，将目标视频帧输入画质增强模型进行处理，结合解码信息和目标视频帧的画质增强数据快速构建非目标视频帧的画质增强效果，使得非目标视频帧的画质增强效果具有画质增强模型处理时的等同效果，并且只需更少的处理时间，这样大大加快了视频的处理速度，因此能够适用于直播视频这类有实时处理需求的场景。

技术领域

本说明书涉及计算机技术领域，尤其涉及一种视频画质增强的方法、装置、设备以及存储介质。

背景技术

画质是指画面质量，图像在压缩过程中如果损失了图像信息，则其画面质量也相应的降低了。画质增强的基本原理是：通过对图像进行分析，确定图像中的色彩信息、物体边缘等数据，通过增强算法对这些数据进行一定尺度的增强，包括锐化物体边缘，提升对比度、饱和度等，使得图像画质得到提升。然而目前的画质增强方法在处理视频时，效率不足，很难适用于一些具有实时需求的场景。

发明内容

为克服相关技术中存在的问题，本说明书提供了一种视频画质增强的方法、装置、设备以及存储介质。

根据本说明书实施例的第一方面，提供一种视频画质增强的方法，包括：

从视频的解码数据中获取视频帧集合和解码信息，所述视频帧集合包括目标视频帧和至少一帧非目标视频帧，所述目标视频帧与所述非目标视频帧之间的时序差值小于或等于预设值；

针对所述视频帧集合，执行如下处理：

将目标视频帧输入训练好的画质增强模型，利用所述画质增强模型获取画质增强后的目标视频帧以及所述目标视频帧的画质增强数据；

基于所述解码信息和所述目标视频帧的画质增强数据，预测出每个所述非目标视频帧的画质增强数据，并利用所述非目标视频帧的画质增强数据对所述非目标视频帧进行画质增强处理。

在某些例子中，上述利用所述画质增强模型获取画质增强后的目标视频帧以及所述目标视频帧的画质增强数据，包括：

获取所述画质增强模型输出的画质增强后的目标视频帧；

根据所述目标视频帧在画质增强前后的对比，得到目标视频帧的画质增强数据。

在某些例子中，上述非目标视频帧包括至少一个帧间块，所述解码信息包括帧间信息，所述帧间信息包括所述帧间块的运动矢量；

所述非目标视频帧的帧间块的画质增强数据基于以下步骤得到：

基于运动矢量查找所述目标视频帧中与所述非目标视频帧相匹配的帧间块，将所述目标视频帧中帧间块的画质增强数据作为所述非目标视频帧中帧间块的画质增强数据。

在某些例子中，上述非目标视频帧还包括至少一个帧内块，所述解码信息还包括帧内信息；

所述非目标视频帧的帧内块的画质增强数据基于所述帧内信息以及所述非目标视频帧中帧间块的画质增强数据预测得到。

在某些例子中，上述帧内信息包括解码过程中利用所述帧内块的周围图像块来预测所述帧内块的相关参数，所述相关参数包括所述周围图像块的抽样值、抽样值的填充方向以及抽样值的加权系数；

所述非目标视频帧的帧内块的画质增强数据基于以下步骤得到：

基于所述相关参数中的抽样值以及所述非目标视频帧中帧间块的画质增强数据，得到所述非目标视频帧中帧内块的周围图像块的画质增强数据的抽样值；

将所述相关参数中的抽样值的填充方向和加权系数，作为画质增强数据的抽样值的填充方向和加权系数，利用所述画质增强数据的抽样值和加权系数进行加权计算，并按照画质增强数据的抽样值的填充方向进行填充，得到所述帧内块的画质增强数据。