dianna.utils.onnx_runner

Module Contents

Classes

SimpleModelRunner

Runs an onnx model with a set of inputs and outputs.
class dianna.utils.onnx_runner.SimpleModelRunner(filename, preprocess_function=None)[source]

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

__call__(input_data)[source]

Get ONNX predictions.