c++ - 将输出帧保存为图像文件 CUDA 解码器

Question

我正在尝试使用 CUDA 解码器项目中的代码将解码后的图像文件保存回 BMP 图像。

            if (g_bReadback && g_ReadbackSID)
            {
                CUresult result = cuMemcpyDtoHAsync(g_bFrameData[active_field], pDecodedFrame[active_field], (nDecodedPitch * nHeight * 3 / 2), g_ReadbackSID);



                long padded_size = (nWidth * nHeight * 3 );
                CString output_file;
                output_file.Format(_T("image/sample_45.BMP"));
                SaveBMP(g_bFrameData[active_field],nWidth,nHeight,padded_size,output_file );

                if (result != CUDA_SUCCESS)
                {
                    printf("cuMemAllocHost returned %d\n", (int)result);
                }
            }

但是保存的图片是这样的

任何人都可以帮我解决我做错了什么..谢谢。

Answer 1

在进一步调查之后，我对您的方法进行了一些修改。

• pDecodedFrame 实际上是一些非 RGB 格式，我认为它是 NV12 格式，我相信它是 particular YUV variant .
• pDecodedFrame 使用特定的 CUDA 内核在 GPU 上转换为 RGB 格式
• 如果指定了 g_bUseInterop，此转换的目标缓冲区将是 OpenGL 提供的表面，或者如果指定了 cudaMalloc 的驱动程序 API 版本，则为普通区域分配的普通区域未指定互操作。

上面提到的目标缓冲区是pInteropFrame(即使在非互操作的情况下)。因此，为您举个例子，为简单起见，我选择仅使用非互操作情况，因为在这种情况下获取 RGB 缓冲区 (pInteropFrame) 会容易得多。

在 cudaPostProcessFrame 用适当的 RGB 图像填充后，此处的方法将 pInteropFrame 复制回主机。还有一个将图像保存为位图文件的例程。我的所有修改都用包含 RMC 的注释描述，所以如果您想找到我所做的所有更改/添加，请搜索它。

要使用，请将此文件放入 cudaDecodeGL项目作为 videoDecodeGL.cpp 源文件的替代品。然后重建项目。然后正常运行可执行文件以显示视频。要捕获特定帧，请使用 nointerop 命令行开关运行可执行文件，例如。 cudaDecodGL nointerop 视频不会显示，但会进行解码操作和帧捕获，帧将保存在 framecap.bmp 文件中。如果要更改捕获的特定帧编号，请将 g_FrameCapSelect = 37; 变量修改为 37 以外的其他数字，然后重新编译。

Here is the replacement对于 videoDecodeGL.cpp，我使用了 pastebin，因为 SO 对问题正文中可以输入的字符数有限制。

请注意，我的方法与是否指定回读无关。我建议不要对此序列使用回读。