vmaf 安装和使用

https://streaminglearningcenter.com/blogs/installing-and-using-netflix-vmaf-master.html

Prepping the Files

Note that vmafossexec.exe only works with YUV inputs so you’ll have to convert your reference and distorted files to YUV format. FFmpeg is the tool of choice and you can modify this command string for your conversion.

ffmpeg -y -i distorted.mp4 -pix_fmt yuv420p -vsync 0 distorted.yuv
ffmpeg -y -i reference.mp4 -pix_fmt yuv420p -vsync 0 reference.yuv

Obviously, you should substitute the actual extension for MP4 if necessary, and if FFmpeg doesn’t decode the file find a different decoder to do so. Note the pixel format used since you’ll input that into your VMAF-Master command string.

Here’s the command string I used during my recent project. Note that both distorted and reference files are in the same folder which also contains the batch file so there are no paths to either file. Here I’m calling the default model, computing VMAF, PSNR, SSIM, and MS SSIM, and storing the results in a file named distorted.csv, an XML-formatted file with a .csv extension.

C:\vmaf-master\x64\Release>vmafossexec yuv420p 1920 820 01-4.0m.yuv 01-2.0m.yuv C:\vmaf-master\model\vmaf_v0.6.1.pkl --psnr --ssim --ms-ssim
Start calculating VMAF score…
Exec FPS: 8.178312
VMAF score = 39.276684
PSNR score = 21.952593
SSIM score = 0.673477
MS-SSIM score = 0.703873

C:\vmaf-master\x64\Release>vmafossexec yuv420p 1920 820 01-4.0m.yuv 01-1.5m.yuv C:\vmaf-master\model\vmaf_v0.6.1.pkl --psnr --ssim --ms-ssim
Start calculating VMAF score…
Exec FPS: 8.109136
VMAF score = 37.880867
PSNR score = 21.920033
SSIM score = 0.674751
MS-SSIM score = 0.704895

C:\vmaf-master\x64\Release>vmafossexec yuv420p 1920 820 01-4.0m.yuv 01-1.0m.yuv C:\vmaf-master\model\vmaf_v0.6.1.pkl --psnr --ssim --ms-ssim
Start calculating VMAF score…
Exec FPS: 7.393184
VMAF score = 34.884087
PSNR score = 21.853622
SSIM score = 0.674992
MS-SSIM score = 0.705316