Crate cervo_onnx

Crate cervo_onnx 

Source
Expand description

Contains utilities for using cervo with ONNX.

§Loading an inference model

use cervo_core::prelude::InfererExt;

let model_data = load_bytes("model.onnx");
let model = cervo_onnx::builder(model_data)
    .build_memoizing(&[])?
    .with_default_epsilon("epsilon");

§Converting to NNEF

use cervo_core::prelude::InfererExt;

let mut onnx_data = load_bytes("model.onnx");
let nnef_data = cervo_onnx::to_nnef(&mut onnx_data, None);

Structs§

OnnxData
Wrapper for a reader providing ONNX data.

Functions§

builder
Utility function for creating an InfererBuilder for OnnxData.
to_nnef
Convert an ONNX model to a NNEF model.