7

我正在尝试从 mp4 视频流中提取图像。在查找内容之后,似乎正确的方法是在 C++ 中使用媒体基础并打开框架/从中读取内容。

文档和示例很少,但经过一番挖掘,似乎有些人通过将帧读入纹理并将该纹理的内容复制到内存可读纹理(我什至不是确定我是否在这里使用了正确的术语)。尝试我发现的东西会给我带来错误,我可能做错了很多事情。

这是我尝试执行此操作的一小段代码(项目本身附在底部)。

    ComPtr<ID3D11Texture2D> spTextureDst;
    MEDIA::ThrowIfFailed(
        m_spDX11SwapChain->GetBuffer(0, IID_PPV_ARGS(&spTextureDst))
        );

    auto rcNormalized = MFVideoNormalizedRect();
    rcNormalized.left = 0;
    rcNormalized.right = 1;
    rcNormalized.top = 0;
    rcNormalized.bottom = 1;
    MEDIA::ThrowIfFailed(
        m_spMediaEngine->TransferVideoFrame(m_spRenderTexture.Get(), &rcNormalized, &m_rcTarget, &m_bkgColor)
        );

    //copy the render target texture to the readable texture.
    m_spDX11DeviceContext->CopySubresourceRegion(m_spCopyTexture.Get(),0,0,0,0,m_spRenderTexture.Get(),0,NULL);
    m_spDX11DeviceContext->Flush();

    //Map the readable texture;                 
    D3D11_MAPPED_SUBRESOURCE mapped = {0};
    m_spDX11DeviceContext->Map(m_spCopyTexture.Get(),0,D3D11_MAP_READ,0,&mapped);
    void* buffer = ::CoTaskMemAlloc(600 * 400 * 3);
    memcpy(buffer, mapped.pData,600 * 400 * 3);
    //unmap so we can copy during next update.
    m_spDX11DeviceContext->Unmap(m_spCopyTexture.Get(),0);


    // and the present it to the screen
    MEDIA::ThrowIfFailed(
        m_spDX11SwapChain->Present(1, 0)
        );            
}

我得到的错误是:

App1.exe 中 0x76814B32 处的第一次机会异常:Microsoft C++ 异常:Platform::InvalidArgumentException ^ 在内存位置 0x07AFF60C。HRESULT:0x80070057

我不确定如何进一步研究它,因为就像我说的那样,关于它的文档很少。

这是我正在处理的修改后的示例。此问题特定于 WinRT(Windows 8 应用程序)。

4

3 回答 3

6

更新成功!!见底部编辑


部分成功,但可能足以回答您的问题。请继续阅读。

在我的系统上,调试异常显示该OnTimer()函数在尝试调用时失败TransferVideoFrame()。它给出的错误是InvalidArgumentException.

所以,我在谷歌上搜索了一下,我的第一个发现是——显然NVIDIA 驱动程序中存在一个错误——这意味着视频播放似乎在 11 和 10 功能级别时失败。

所以我的第一个变化是功能CreateDX11Device()如下:

static const D3D_FEATURE_LEVEL levels[] = {
/*
        D3D_FEATURE_LEVEL_11_1,
        D3D_FEATURE_LEVEL_11_0,  
        D3D_FEATURE_LEVEL_10_1,
        D3D_FEATURE_LEVEL_10_0,
*/
        D3D_FEATURE_LEVEL_9_3,
        D3D_FEATURE_LEVEL_9_2,
        D3D_FEATURE_LEVEL_9_1
    };

现在TransferVideoFrame()仍然失败,但给出E_FAIL(作为 HRESULT)而不是无效参数。

更多谷歌搜索导致我的第二个发现-

这是一个示例,展示了使用TransferVideoFrame()不使用CreateTexture2D()来预先创建纹理。我看到您已经有一些OnTimer()与此类似但未使用的代码,所以我猜您找到了相同的链接。

无论如何,我现在使用这段代码来获取视频帧:

ComPtr <ID3D11Texture2D> spTextureDst;
m_spDX11SwapChain->GetBuffer (0, IID_PPV_ARGS (&spTextureDst));
m_spMediaEngine->TransferVideoFrame (spTextureDst.Get (), nullptr, &m_rcTarget, &m_bkgColor);

完成此操作后,我看到TransferVideoFrame()成功(好!)但调用Map()您复制的纹理 - m_spCopyTexture- 失败,因为该纹理不是使用 CPU 读取访问权限创建的。

因此,我只是使用您的读/写作m_spRenderTexture为副本的目标,因为它具有正确的标志,并且由于之前的更改,我不再使用它。

        //copy the render target texture to the readable texture.
        m_spDX11DeviceContext->CopySubresourceRegion(m_spRenderTexture.Get(),0,0,0,0,spTextureDst.Get(),0,NULL);
        m_spDX11DeviceContext->Flush();

        //Map the readable texture;                 
        D3D11_MAPPED_SUBRESOURCE mapped = {0};
        HRESULT hr = m_spDX11DeviceContext->Map(m_spRenderTexture.Get(),0,D3D11_MAP_READ,0,&mapped);
        void* buffer = ::CoTaskMemAlloc(176 * 144 * 3);
        memcpy(buffer, mapped.pData,176 * 144 * 3);
        //unmap so we can copy during next update.
        m_spDX11DeviceContext->Unmap(m_spRenderTexture.Get(),0);

现在,在我的系统上,该OnTimer()功能不会失败。视频帧被渲染到纹理,像素数据被成功复制到内存缓冲区。

在查看是否有进一步的问题之前,也许这是一个很好的时机,看看你是否能取得和我目前一样的进步。如果您对此答案发表评论并提供更多信息,我将编辑答案以添加更多帮助(如果可能)。

编辑

对纹理描述的更改FramePlayer::CreateBackBuffers()

        //make first texture cpu readable
        D3D11_TEXTURE2D_DESC texDesc = {0};
        texDesc.Width = 176;
        texDesc.Height = 144;
        texDesc.MipLevels = 1;
        texDesc.ArraySize = 1;
        texDesc.Format = DXGI_FORMAT_B8G8R8A8_UNORM;
        texDesc.SampleDesc.Count = 1;
        texDesc.SampleDesc.Quality =  0;
        texDesc.Usage = D3D11_USAGE_STAGING;
        texDesc.BindFlags = 0;
        texDesc.CPUAccessFlags =  D3D11_CPU_ACCESS_READ | D3D11_CPU_ACCESS_WRITE;
        texDesc.MiscFlags = 0;

        MEDIA::ThrowIfFailed(m_spDX11Device->CreateTexture2D(&texDesc,NULL,&m_spRenderTexture));

另请注意,有时需要清除内存泄漏(我相信您知道) - 在下一行中分配的内存永远不会被释放:

void* buffer = ::CoTaskMemAlloc(176 * 144 * 3); // sizes changed for my test

成功

我现在已经成功保存了一个单独的帧,但现在没有使用复制纹理。

首先,我下载了最新版本的 DirectXTex 库,它提供了 DX11 纹理辅助函数,例如从纹理中提取图像并保存到文件。需要仔细遵循将 DirectXTex 库作为现有项目添加到您的解决方案的说明,并注意 Windows 8 应用商店应用所需的更改

一旦包含、引用和构建了上述库,将以下内容添加#includeFramePlayer.cpp

#include "..\DirectXTex\DirectXTex.h"  // nb - use the relative path you copied to
#include <wincodec.h>

最后,代码的中心部分FramePlayer::OnTimer()需要类似于以下内容。你会看到我每次都保存到相同的文件名,所以这需要修改,例如在名称中添加帧号

// new frame available at the media engine so get it 
ComPtr<ID3D11Texture2D> spTextureDst;
MEDIA::ThrowIfFailed(m_spDX11SwapChain->GetBuffer(0, IID_PPV_ARGS(&spTextureDst)));

auto rcNormalized = MFVideoNormalizedRect();
rcNormalized.left = 0;
rcNormalized.right = 1;
rcNormalized.top = 0;
rcNormalized.bottom = 1;
MEDIA::ThrowIfFailed(m_spMediaEngine->TransferVideoFrame(spTextureDst.Get(), &rcNormalized, &m_rcTarget, &m_bkgColor));

// capture an image from the DX11 texture
DirectX::ScratchImage pImage;
HRESULT hr = DirectX::CaptureTexture(m_spDX11Device.Get(), m_spDX11DeviceContext.Get(), spTextureDst.Get(), pImage);
if (SUCCEEDED(hr))
{
    // get the image object from the wrapper
    const DirectX::Image *pRealImage = pImage.GetImage(0, 0, 0);

    // set some place to save the image frame
    StorageFolder ^dataFolder = ApplicationData::Current->LocalFolder;
    Platform::String ^szPath = dataFolder->Path + "\\frame.png";

    // save the image to file
    hr = DirectX::SaveToWICFile(*pRealImage, DirectX::WIC_FLAGS_NONE, GUID_ContainerFormatPng, szPath->Data());
}

// and the present it to the screen
MEDIA::ThrowIfFailed(m_spDX11SwapChain->Present(1, 0));            

I don't have time right now to take this any further but I'm very pleased with what I have achieved so far :-))

Can you take a fresh look and update your results in comments?

于 2013-04-08T08:16:56.753 回答
0

查看Video Thumbnail SampleSource Reader文档。

您可以在下面找到示例代码SDK Root\Samples\multimedia\mediafoundation\VideoThumbnail

于 2013-04-05T11:59:16.647 回答
-2

我认为 OpenCV 可以帮助你。OpenCV 提供 api 来从相机或视频文件中捕获帧。你可以在这里下载它http://opencv.org/downloads.html。以下是我用“OpenCV 2.3.1”编写的演示。

#include "opencv.hpp"

using namespace cv;
int main()
{
    VideoCapture cap("demo.avi");   // open a video to capture
    if (!cap.isOpened())        // check if succeeded
        return -1;

    Mat frame;
    namedWindow("Demo", CV_WINDOW_NORMAL);
    // Loop to capture frame and show on the window
    while (1) {
        cap >> frame;
        if (frame.empty())
            break;

        imshow("Demo", frame);
        if (waitKey(33) >= 0)  // pause 33ms every frame
            break;
    }

    return 0;
}
于 2013-04-06T04:32:50.053 回答