Class ONNXModelConverter
ONNX model converter to Barracuda format.
Namespace: Unity.Barracuda.ONNX
Syntax
public class ONNXModelConverter : objectConstructors
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. |
Events
ModelImported
Model imported event
Declaration
public static event Action<object, Model> ModelImportedEvent Type
| Type | Description |
|---|---|
| Action<Object, Model> |