Add vmaf --cuda option #178
Conversation
| "[0:v]scale_cuda=format={pix_fmt},scale_cuda={w}:{h},setpts=PTS-STARTPTS[dis];\ | ||
| [1:v]scale_cuda=format={pix_fmt},scale_cuda={w}:{h},{ref_vf},setpts=PTS-STARTPTS[ref];[dis][ref]" | ||
| ) | ||
| } else { | ||
| format!( | ||
| "[0:v]scale_cuda=format={pix_fmt},setpts=PTS-STARTPTS[dis];\ | ||
| [1:v]scale_cuda=format={pix_fmt},{ref_vf}setpts=PTS-STARTPTS[ref];[dis][ref]" |
There was a problem hiding this comment.
Is this correct for cuda vmaf?
There was a problem hiding this comment.
the vmaf documentation isnt a great help here... maybe this?
let prefix = if let Some((w, h)) = self.vf_scale(model.unwrap_or_default(), distorted_res) {
format!(
"[0:v]scale_cuda=format={pix_fmt},scale={w}:{h},setpts=PTS-STARTPTS[dis];
[1:v]scale_cuda=format={pix_fmt},scale={w}:{h},{ref_vf}setpts=PTS-STARTPTS[ref];[dis][ref]"
)
} else {
format!(
"[0:v]scale_cuda=format={pix_fmt},setpts=PTS-STARTPTS[dis];
[1:v]scale_cuda=format={pix_fmt},{ref_vf}setpts=PTS-STARTPTS[ref];[dis][ref]"
)
};
|
So I had a chance to test this version and was able to confirm that some things works and some don't. What worksRunning For example, av1 => av1, x264 => x264, x265 => x265. What doesn't workWhenever the reference and distorted content doesn't use the same codec. For example, x264 => av1, x265 => av1, x264 => x265 How to reproduceI can reproduce this using the "Big Buck Bunny" test video:
The command will fail with the following output: Details
DEBUG: Using ffmpeg -filter_complex [0:v]scale_cuda=format=yuv420p10le,setpts=PTS-STARTPTS[dis];[1:v]scale_cuda=format=yuv420p10le,setpts=PTS-STARTPTS[ref];[dis][ref]libvmaf_cuda=
|
Add support for CUDA accelerated vmaf. See #163.
Run vmaf command with arg
--cudato enable.This is experimental/incomplete. I can't test this properly myself as I lack nvidia hardware.