中文

基于 PyTorch 的 VMAF 重实现:若干实验结果

机器学习 2024-07-17 v4 计算机视觉与模式识别

摘要

基于标准 VMAF 实现,我们提出了一种使用 PyTorch 框架的 VMAF 实现。该实现与标准实现(libvmaf)的比较显示 VMAF 单位下的偏差 102\lesssim 10^{-2}。我们研究了将 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