Class ONNXModelConverter
ONNX model converter to Barracuda format.
Inherited Members
Namespace: Unity.Barracuda.ONNX
Assembly: Unity.Barracuda.ONNX.dll
Syntax
public class ONNXModelConverterConstructors
ONNXModelConverter(bool, bool, bool)
Constructs ONNX model converter
Declaration
public ONNXModelConverter(bool optimizeModel, bool treatErrorsAsWarnings = false, bool forceArbitraryBatchSize = true)Parameters
| Type | Name | Description |
|---|---|---|
| bool | optimizeModel | Enable/disable various model optimizations while importing model from ONNX format. |
| bool | treatErrorsAsWarnings | Treat import errors as warnings. |
| bool | 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. |
Events
ModelImported
Model imported event
Declaration
public static event Action<object, Model> ModelImportedEvent Type
| Type | Description |
|---|---|
| Action<object, Model> |