onnxruntime/winml/test/image/imageTestHelper.h
Sheil Kumar 2717c178cc
Fork the WinML APIs into the Microsoft namespace (#3503)
* Migrate winml to Microsoft Namespace (packaging changes are pending)

* add ns_prefix toggle

* fix packaging

* Users/sheilk/add missing raw header (#3484)

* add dualapipartition

* wrong variable for repo root

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>

* remove existence check to force failures

* extra paren

* dualapipartition needs to be referenced from the source

* add microsoft.ai.machinelearning.dll to the output dir

* rename the idl file so that assembly info is correctly added into the winmd

* fix namespaces

* update namespaces

* default to microsoft, and add namespace override as build argument

* update cmakesetings.json as well

* remove from cmakelists.txt

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
Co-authored-by: Changming Sun <chasun@microsoft.com>
2020-04-17 06:18:54 -07:00

34 lines
1.2 KiB
C++

//-----------------------------------------------------------------------------
//
// 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);
}