23 instantiations of OnnxScoringEstimator
Microsoft.ML.DnnImageFeaturizer.AlexNet (2)
AlexNetExtension.cs (2)
46
var prepEstimator = new
OnnxScoringEstimator
(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "AlexNetPrepOnnx", "AlexNetPreprocess.onnx"));
47
var mainEstimator = new
OnnxScoringEstimator
(env, new[] { "Dropout234_Output_0" }, new[] { "Input140" }, Path.Combine(modelDir, "AlexNetOnnx", "AlexNet.onnx"));
Microsoft.ML.DnnImageFeaturizer.ResNet101 (2)
ResNet101Extension.cs (2)
46
var prepEstimator = new
OnnxScoringEstimator
(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "ResNetPrepOnnx", "ResNetPreprocess.onnx"));
47
var mainEstimator = new
OnnxScoringEstimator
(env, new[] { "Pooling2286_Output_0" }, new[] { "Input1600" }, Path.Combine(modelDir, "ResNet101Onnx", "ResNet101.onnx"));
Microsoft.ML.DnnImageFeaturizer.ResNet18 (2)
ResNet18Extension.cs (2)
46
var prepEstimator = new
OnnxScoringEstimator
(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "ResNetPrepOnnx", "ResNetPreprocess.onnx"));
47
var mainEstimator = new
OnnxScoringEstimator
(env, new[] { "Pooling395_Output_0" }, new[] { "Input247" }, Path.Combine(modelDir, "ResNet18Onnx", "ResNet18.onnx"));
Microsoft.ML.DnnImageFeaturizer.ResNet50 (2)
ResNet50Extension.cs (2)
46
var prepEstimator = new
OnnxScoringEstimator
(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "ResNetPrepOnnx", "ResNetPreprocess.onnx"));
47
var mainEstimator = new
OnnxScoringEstimator
(env, new[] { "Pooling1096_Output_0" }, new[] { "Input750" }, Path.Combine(modelDir, "ResNet50Onnx", "ResNet50.onnx"));
Microsoft.ML.OnnxTransformer (15)
OnnxCatalog.cs (15)
44
return new
OnnxScoringEstimator
(env, modelFile, gpuDeviceIdToUse, fallbackToCpuToUse);
75
return new
OnnxScoringEstimator
(env, modelBytes, gpuDeviceIdToUse, fallbackToCpuToUse);
114
return new
OnnxScoringEstimator
(env, modelFile, gpuDeviceIdToUse, fallbackToCpuToUse, shapeDictionary: shapeDictionary);
153
return new
OnnxScoringEstimator
(env, modelBytes, gpuDeviceIdToUse, fallbackToCpuToUse, shapeDictionary: shapeDictionary);
188
return new
OnnxScoringEstimator
(env, new[] { outputColumnName }, new[] { inputColumnName }, modelFile, gpuDeviceIdToUse, fallbackToCpuToUse);
223
return new
OnnxScoringEstimator
(env, new[] { outputColumnName }, new[] { inputColumnName }, modelBytes, gpuDeviceIdToUse, fallbackToCpuToUse);
242
return new
OnnxScoringEstimator
(env, options.OutputColumns, options.InputColumns, options.ModelFile,
283
return new
OnnxScoringEstimator
(env, new[] { outputColumnName }, new[] { inputColumnName },
324
return new
OnnxScoringEstimator
(env, new[] { outputColumnName }, new[] { inputColumnName },
353
return new
OnnxScoringEstimator
(env, outputColumnNames, inputColumnNames, modelFile, gpuDeviceIdToUse, fallbackToCpuToUse);
381
return new
OnnxScoringEstimator
(env, outputColumnNames, inputColumnNames, modelBytes, gpuDeviceIdToUse, fallbackToCpuToUse);
414
return new
OnnxScoringEstimator
(env, outputColumnNames, inputColumnNames, modelFile, gpuDeviceIdToUse, fallbackToCpuToUse, shapeDictionary: shapeDictionary);
447
return new
OnnxScoringEstimator
(env, outputColumnNames, inputColumnNames, modelBytes, gpuDeviceIdToUse, fallbackToCpuToUse, shapeDictionary: shapeDictionary);
482
return new
OnnxScoringEstimator
(env, outputColumnNames, inputColumnNames, modelFile, gpuDeviceIdToUse, fallbackToCpuToUse, shapeDictionary: shapeDictionary, recursionLimit);
517
return new
OnnxScoringEstimator
(env, outputColumnNames, inputColumnNames, modelBytes, gpuDeviceIdToUse, fallbackToCpuToUse, shapeDictionary: shapeDictionary, recursionLimit);
90 references to OnnxScoringEstimator
Microsoft.ML.DnnImageFeaturizer.AlexNet (2)
AlexNetExtension.cs (2)
46
var
prepEstimator = new OnnxScoringEstimator(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "AlexNetPrepOnnx", "AlexNetPreprocess.onnx"));
47
var
mainEstimator = new OnnxScoringEstimator(env, new[] { "Dropout234_Output_0" }, new[] { "Input140" }, Path.Combine(modelDir, "AlexNetOnnx", "AlexNet.onnx"));
Microsoft.ML.DnnImageFeaturizer.ResNet101 (2)
ResNet101Extension.cs (2)
46
var
prepEstimator = new OnnxScoringEstimator(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "ResNetPrepOnnx", "ResNetPreprocess.onnx"));
47
var
mainEstimator = new OnnxScoringEstimator(env, new[] { "Pooling2286_Output_0" }, new[] { "Input1600" }, Path.Combine(modelDir, "ResNet101Onnx", "ResNet101.onnx"));
Microsoft.ML.DnnImageFeaturizer.ResNet18 (2)
ResNet18Extension.cs (2)
46
var
prepEstimator = new OnnxScoringEstimator(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "ResNetPrepOnnx", "ResNetPreprocess.onnx"));
47
var
mainEstimator = new OnnxScoringEstimator(env, new[] { "Pooling395_Output_0" }, new[] { "Input247" }, Path.Combine(modelDir, "ResNet18Onnx", "ResNet18.onnx"));
Microsoft.ML.DnnImageFeaturizer.ResNet50 (2)
ResNet50Extension.cs (2)
46
var
prepEstimator = new OnnxScoringEstimator(env, new[] { "PreprocessedInput" }, new[] { "OriginalInput" }, Path.Combine(modelDir, "ResNetPrepOnnx", "ResNetPreprocess.onnx"));
47
var
mainEstimator = new OnnxScoringEstimator(env, new[] { "Pooling1096_Output_0" }, new[] { "Input750" }, Path.Combine(modelDir, "ResNet50Onnx", "ResNet50.onnx"));
Microsoft.ML.IntegrationTests (3)
ONNX.cs (3)
60
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(modelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
106
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(modelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
158
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(modelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
Microsoft.ML.OnnxTransformer (52)
DnnImageFeaturizerTransform.cs (2)
25
/// a chain of two <see cref="
OnnxScoringEstimator
"/>.
92
/// <see cref="
OnnxScoringEstimator
"/>s (one for preprocessing and one with a pretrained image DNN) with specific models
OnnxCatalog.cs (47)
17
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the input column.
19
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
41
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog, string modelFile, int? gpuDeviceId = null, bool fallbackToCpu = false)
48
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the input column.
50
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
72
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog, Stream modelBytes, int? gpuDeviceId = null, bool fallbackToCpu = false)
79
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the input column.
81
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
107
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
118
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the input column.
120
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
146
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
157
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnName"/> column.
158
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
180
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
192
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnName"/> column.
193
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
215
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
227
/// Create a <see cref="
OnnxScoringEstimator
"/> using the specified <see cref="OnnxOptions"/>.
228
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
238
/// <param name="options">Options for the <see cref="
OnnxScoringEstimator
"/>.</param>
239
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog, OnnxOptions options)
247
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnName"/> column.
248
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
274
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
288
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnName"/> column.
289
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
315
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
329
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnNames"/> columns.
330
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
345
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
357
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnNames"/> columns.
358
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
373
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
385
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnNames"/> columns.
386
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
405
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
418
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnNames"/> columns.
419
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
438
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
451
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnNames"/> columns.
452
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
472
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
486
/// Create a <see cref="
OnnxScoringEstimator
"/>, which applies a pre-trained Onnx model to the <paramref name="inputColumnNames"/> columns.
487
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
507
public static
OnnxScoringEstimator
ApplyOnnxModel(this TransformsCatalog catalog,
527
/// <see cref="
OnnxScoringEstimator
"/> (one for preprocessing and one with a pretrained image DNN) with specific models
OnnxOptions.cs (1)
10
/// The options for an <see cref="
OnnxScoringEstimator
"/>.
OnnxTransform.cs (2)
38
/// <see cref="ITransformer"/> resulting from fitting an <see cref="
OnnxScoringEstimator
"/>.
39
/// Please refer to <see cref="
OnnxScoringEstimator
"/> to learn more about the necessary dependencies,
Microsoft.ML.OnnxTransformerTest (13)
OnnxTransformTests.cs (13)
156
var
pipe = useOptionsCtor ?
196
var
est = ML.Transforms.ApplyOnnxModel(outputNames, inputNames, modelFile, gpuDeviceId, fallbackToCpu);
369
var
pipeline = ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
408
var
pipeline = ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
436
var
pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outa", "outb" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
476
var
pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outb", "outa" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
536
var
pipeline = ML.Transforms.ApplyOnnxModel(modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
562
var
pipeline = ML.Transforms.ApplyOnnxModel(modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
694
var
pipeline = ML.Transforms.ApplyOnnxModel(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
747
var
pipeline = ML.Transforms.ApplyOnnxModel(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
902
var pipeline = new
OnnxScoringEstimator
[3];
972
var
pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outa", "outb" }, new[] { "ina", "inb" },
1072
var
pipeline = ML.Transforms.ApplyOnnxModel(nameof(PredictionWithCustomShape.argmax),
Microsoft.ML.Samples (2)
Dynamic\ModelOperations\OnnxConversion.cs (1)
84
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(onnxModelPath);
Dynamic\Transforms\ApplyOnnxModel.cs (1)
28
var
pipeline = mlContext.Transforms.ApplyOnnxModel(modelPath);
Microsoft.ML.Tests (12)
OnnxConversionTest.cs (10)
62
/// call <see cref="
OnnxScoringEstimator
"/> to evaluate that file. The outputs of <see cref="
OnnxScoringEstimator
"/> are checked against the original
826
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(onnxModelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
988
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(onnxModelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
1515
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(onnxFilePath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
1632
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(outputNames, inputNames, onnxModelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
1743
var
pipeline = mlContext.Transforms.ApplyOnnxModel(onnxModelPath);
1806
var
onnxEstimator = mlContext.Transforms.ApplyOnnxModel(outputNames, inputNames, onnxModelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
1819
var
onnxEstimator2 = mlContext.Transforms.ApplyOnnxModel(outputNames, inputNames, onnxModelPath2, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
2254
var
onnxEstimator = ML.Transforms.ApplyOnnxModel(onnxModelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
OnnxSequenceTypeWithAttributesTest.cs (2)
46
var
pipeline = ctx.Transforms.ApplyOnnxModel(
85
var
pipeline = ctx.Transforms.ApplyOnnxModel(