1

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

但是保存的图像看起来像这样在此处输入图像描述

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

4

1 回答 1

3

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

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

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

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

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

这是我使用 pastebin 的替代品，videoDecodeGL.cpp 因为 SO 对可以在问题正文中输入的字符数有限制。

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

于 2013-08-24T21:33:51.357 回答