dianna.utils.onnx_runner

Classes

SimpleModelRunner

Runs an onnx model with a set of inputs and outputs.

Module Contents

class dianna.utils.onnx_runner.SimpleModelRunner(filename, preprocess_function=None)[source]

Runs an onnx model with a set of inputs and outputs.

filename[source]

preprocess_function[source]

__call__(input_data)[source]: Get ONNX predictions.