基于 PyTorch 的 VMAF 重实现:若干实验结果
机器学习
2024-07-17 v4 计算机视觉与模式识别
摘要
基于标准 VMAF 实现,我们提出了一种使用 PyTorch 框架的 VMAF 实现。该实现与标准实现(libvmaf)的比较显示 VMAF 单位下的偏差 。我们研究了将 VMAF 用作目标函数时的梯度计算,并证明使用该函数的训练不会产生异常的梯度。该实现随后被用于训练一个预处理滤波器。结果表明其性能优于非锐化掩码滤波器。所得滤波器也易于实现,并可应用于视频处理任务中以改善视频压缩。数值实验结果证实了这一点。
引用
@article{arxiv.2310.15578,
title = {VMAF Re-implementation on PyTorch: Some Experimental Results},
author = {Kirill Aistov and Maxim Koroteev},
journal= {arXiv preprint arXiv:2310.15578},
year = {2024}
}
备注
7 pages, 3 figures