mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
2717c178cc
* 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>
98 lines
No EOL
2.1 KiB
C++
98 lines
No EOL
2.1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include "ImageFeatureDescriptor.g.h"
|
|
|
|
namespace WINMLP {
|
|
|
|
enum class ImageNominalPixelRange {
|
|
ImageNominalPixelRange_NominalRange_0_255,
|
|
ImageNominalPixelRange_Normalized_0_1,
|
|
ImageNominalPixelRange_Normalized_1_1,
|
|
ImageNominalPixelRange_NominalRange_16_235,
|
|
};
|
|
enum class ImageColorSpaceGamma {
|
|
ImageColorSpaceGamma_Linear,
|
|
ImageColorSpaceGamma_SRGB,
|
|
};
|
|
|
|
struct ImageFeatureDescriptor : ImageFeatureDescriptorT<
|
|
ImageFeatureDescriptor,
|
|
ILearningModelFeatureDescriptorNative> {
|
|
ImageFeatureDescriptor() = delete;
|
|
ImageFeatureDescriptor(
|
|
const char* name,
|
|
const char* description,
|
|
winml::TensorKind tensor_kind,
|
|
const std::vector<int64_t>& shape,
|
|
bool is_required,
|
|
wgi::BitmapPixelFormat pixelformat,
|
|
wgi::BitmapAlphaMode alphamode,
|
|
uint32_t width,
|
|
uint32_t height,
|
|
ImageNominalPixelRange nominalPixelRange,
|
|
ImageColorSpaceGamma colorSpaceGamma);
|
|
|
|
wgi::BitmapPixelFormat
|
|
BitmapPixelFormat();
|
|
|
|
wgi::BitmapAlphaMode
|
|
BitmapAlphaMode();
|
|
|
|
uint32_t
|
|
Width();
|
|
|
|
uint32_t
|
|
Height();
|
|
|
|
hstring
|
|
Name();
|
|
|
|
hstring
|
|
Description();
|
|
|
|
winml::LearningModelFeatureKind
|
|
Kind();
|
|
|
|
bool
|
|
IsRequired();
|
|
|
|
winml::TensorKind
|
|
TensorKind();
|
|
|
|
wfc::IVectorView<int64_t>
|
|
Shape();
|
|
|
|
ImageNominalPixelRange
|
|
GetNominalPixelRange();
|
|
|
|
ImageColorSpaceGamma
|
|
GetColorSpaceGamma();
|
|
|
|
STDMETHOD(GetName)
|
|
(
|
|
const wchar_t** name,
|
|
uint32_t* cchName) override;
|
|
|
|
STDMETHOD(GetDescription)
|
|
(
|
|
const wchar_t** description,
|
|
uint32_t* cchDescription) override;
|
|
|
|
private:
|
|
winrt::hstring name_;
|
|
winrt::hstring description_;
|
|
winml::TensorKind tensor_kind_;
|
|
std::vector<int64_t> shape_;
|
|
bool is_required_;
|
|
wgi::BitmapPixelFormat pixel_format_;
|
|
wgi::BitmapAlphaMode alpha_mode_;
|
|
uint32_t width_;
|
|
uint32_t height_;
|
|
ImageNominalPixelRange nominal_pixel_range_;
|
|
ImageColorSpaceGamma color_space_gamma_;
|
|
};
|
|
|
|
} // namespace WINMLP
|