Class ONNXModelConverter

ONNX model converter to Barracuda format.

Inheritance

Object

ONNXModelConverter

Namespace: Unity.Barracuda.ONNX

Syntax

public class ONNXModelConverter : object

Constructors

ONNXModelConverter(Boolean, Boolean, Boolean)

Constructs ONNX model converter

Declaration

public ONNXModelConverter(bool optimizeModel, bool treatErrorsAsWarnings = false, bool forceArbitraryBatchSize = true)

Parameters

Type	Name	Description
Boolean	optimizeModel	Enable/disable various model optimizations while importing model from ONNX format.
Boolean	treatErrorsAsWarnings	Treat import errors as warnings.
Boolean	forceArbitraryBatchSize	Repair model input batch size. Sometimes needed for ONNX models coming from PyTorch.

Methods

Convert(Byte[])

Convert ONNX model and return Barracuda Model object.

Declaration

public Model Convert(byte[] buffer)

Parameters

Type	Name	Description
Byte[]	buffer	Memory buffer containing ONNX model.

Returns

Type	Description
Model	Barracuda Model object.

Convert(String)

Convert ONNX model and return Barracuda Model object.

Declaration

public Model Convert(string filePath)

Parameters

Type	Name	Description
String	filePath	Location of the input ONNX model.

Returns

Type	Description
Model	Barracuda Model object.