基于卷积神经网络的环路滤波方法和装置

摘要

本发明公开了一种基于卷积神经网络的环路滤波方法和装置。该方法的步骤包括：建立并训练用于重建视频质量恢复的卷积神经网络，包括第一网络、第二网络和第三网络；利用索贝尔算子和二阶拉普拉斯算子求出输入的重建图像的一阶梯度和二阶梯度；利用第一网络和第三网络对输入的重建图像的一阶梯度和二阶梯度进行恢复，并将恢复的梯度与输入的重建图像的特征图进行整合，得到整合后的特征图；将整合后的特征图经过第二网络，得到恢复后的重建图像。相对于传统的环路滤波技术，本发明通过卷积网络提取视频特征，提高了对压缩视频的恢复能力。本发明在GPU的加速下能够与传统滤波技术的计算复杂度相当，且能够更加有效提升视频编码器压缩视频质量。