ATLAS Offline Software
Loading...
Searching...
No Matches
AthExOnnxRuntime_test_infer Namespace Reference

Functions

 AthExOnnxRuntimeExampleCfg (flags, name="AthOnnxExample", **kwargs)

Variables

 flags = initConfigFlags()
 ExecutionProvider
 acc = MainServicesCfg(flags)
 withDetails
 True
 summariseProps

Function Documentation

◆ AthExOnnxRuntimeExampleCfg()

AthExOnnxRuntime_test_infer.AthExOnnxRuntimeExampleCfg ( flags,
name = "AthOnnxExample",
** kwargs )

Definition at line 9 of file AthExOnnxRuntime_test_infer.py.

9def AthExOnnxRuntimeExampleCfg(flags, name="AthOnnxExample", **kwargs):
10 acc = ComponentAccumulator()
11
12 model_fname = "dev/MLTest/2020-03-02/MNIST_testModel.onnx"
13 execution_provider = OnnxRuntimeType.CPU
14 from AthOnnxComps.OnnxRuntimeInferenceConfig import OnnxRuntimeInferenceToolCfg
15 kwargs.setdefault("ORTInferenceTool", acc.popToolsAndMerge(
16 OnnxRuntimeInferenceToolCfg(flags, model_fname, execution_provider)
17 ))
18
19 input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
20 kwargs.setdefault("BatchSize", 3)
21 kwargs.setdefault("InputDataPixel", input_data)
22 kwargs.setdefault("OutputLevel", Constants.DEBUG)
23 acc.addEventAlgo(CompFactory.AthOnnx.EvaluateModelWithAthInfer(name, **kwargs))
24
25 return acc
26

Variable Documentation

◆ acc

AthExOnnxRuntime_test_infer.acc = MainServicesCfg(flags)

Definition at line 38 of file AthExOnnxRuntime_test_infer.py.

◆ ExecutionProvider

AthExOnnxRuntime_test_infer.ExecutionProvider

Definition at line 35 of file AthExOnnxRuntime_test_infer.py.

◆ flags

AthExOnnxRuntime_test_infer.flags = initConfigFlags()

Definition at line 34 of file AthExOnnxRuntime_test_infer.py.

◆ summariseProps

AthExOnnxRuntime_test_infer.summariseProps

Definition at line 40 of file AthExOnnxRuntime_test_infer.py.

◆ True

AthExOnnxRuntime_test_infer.True

Definition at line 40 of file AthExOnnxRuntime_test_infer.py.

◆ withDetails

AthExOnnxRuntime_test_infer.withDetails

Definition at line 40 of file AthExOnnxRuntime_test_infer.py.