onnxruntime/winml/test/image/imageTestHelper.h

//-----------------------------------------------------------------------------
//
//  Copyright (c) Microsoft Corporation. All rights reserved.
//
//-----------------------------------------------------------------------------

#pragma once

#include <winrt/Windows.Media.h>
#include <winrt/Windows.Graphics.Imaging.h>

enum OutputBindingStrategy { Bound, Unbound, Empty };
enum EvaluationStrategy { Async, Sync };
enum ModelInputOutputType { Image, Tensor };
enum InputImageSource { FromVideoFrame, FromImageFeatureValue, FromCPUResource, FromGPUResource };
enum VideoFrameSource { FromSoftwareBitmap, FromDirect3DSurface, FromUnsupportedD3DSurface };

namespace ImageTestHelper
{
    wgi::BitmapPixelFormat GetPixelFormat(const std::wstring& inputPixelFormat);

    winml::TensorFloat LoadInputImageFromCPU(
                wgi::SoftwareBitmap softwareBitmap,
                const std::wstring& modelPixelFormat);

    winml::TensorFloat LoadInputImageFromGPU(
                wgi::SoftwareBitmap softwareBitmap,
                const std::wstring& modelPixelFormat);

    bool VerifyHelper(
                wm::VideoFrame actual,
                wm::VideoFrame expected);

}