将输出帧保存为图像文件 CUDA 解码器答案

【问题标题】：Saving output frame as an image file CUDA decoder将输出帧保存为图像文件 CUDA 解码器
【发布时间】：2013-08-23 10:40:31
【问题描述】：

我正在尝试使用 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);
                }
            }

但保存的图像看起来像这样

谁能帮我看看我做错了什么..谢谢。

【问题讨论】：

尝试使用cuMemcpyDtoH 而不是cuMemcpyDtoHAsync，看看是否有任何不同。
不输出还是一样的
在这种情况下，我可以看到您没有 CUDA 编程问题。唯一明显可能出错的是您正在使用异步内存传输，在您开始保存帧之前可能无法完成。但这没有帮助。所以这两种选择要么是——从 GPU 出来的数据已经错误（这可能意味着它被破坏了），要么保存例程期望它接收到的数据与你传递的数据不同。您在此处显示的任何内容都无法解决这两个问题中的任何一个。
数据正在从结果显示正常的窗口中读取。我正在使用 CUDA 解码器 D3D9 示例代码并尝试保存从窗口读取的帧。他们说帧存储在 BYTE* 数组 g_bFrameData[active_field] 中，我试图将其保存为 bmp 文件..

标签： c++ visual-studio-2010 visual-c++ image-processing cuda

【解决方案1】：

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

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

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

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

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

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

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

【讨论】：