如何在不使用命令行的情况下在 GPU 上执行 ffmpeg 代码？

Question

我们用 C 代码编写了一段短代码来读取视频文件，使用常用库如 libavcodec、libavformat 等。

代码运行平稳，但仅使用 CPU 资源。我们需要在 GPU（Nvidia GeForce 940MX 和 1080Ti）上运行代码。有没有办法强制代码在 GPU 上运行？

虽然使用命令行（例如，ffmpeg -hwaccel cuvid -i vid.mp4 out.avi）一切都很好，但我们无法通过源代码让它在 GPU 上运行。

我们正在使用 Ubuntu 18.04，并且 ffmpeg 使用 CUDA 9.2 正确编译

Answer 1

https://github.com/FFmpeg/FFmpeg/tree/master/doc/examples 有使用 libav (ffmpeg) 编码和解码视频的非常好的示例。

您需要的是demuxing_decoding.c 示例并更改166 行，即：

/* find decoder for the stream */
dec = avcodec_find_decoder(st->codecpar->codec_id);

与

/* find decoder for the stream */
if (st->codecpar->codec_id == AV_CODEC_ID_H264)
{
    dec = avcodec_find_decoder_by_name("h264_cuvid");
}
else if (st->codecpar->codec_id == AV_CODEC_ID_HEVC)
{
    dec = avcodec_find_decoder_by_name("hevc_cuvid");
}
else
{
    dec = avcodec_find_decoder(st->codecpar->codec_id);
}

为其他格式添加/更改行。并确保您的 FFmpeg 使用 --enable-cuda --enable-cuvid

编译

在我的测试中，我得到错误来自第 85 行：因为 nvdec (hevc_cuvid) 使用 p010 10 位内部格式（输入为 yuv420p10）。这意味着解码帧将是NV12 像素格式或P010，具体取决于位深度。我希望您熟悉像素格式。

希望对您有所帮助。