mlua-tract

Lua bindings for tract, the tiny, no-nonsense, self-contained, Rust-based Tensorflow and ONNX inference runtime.

Installing

Add to your Rust project using one of MLua's features: [lua51, lua52, lua53, lua54, luajit, luajit52].

$ cargo add mlua-tract --features luajit

Using

use mlua::Lua;
use mlua_tract;

let lua = Lua::new();
mlua_tract::preload(&lua)?;
let script = r#"
    local tract_onnx = require('tract.onnx')
    local model = tract_onnx.model_for_path('testdata/keras-tract-tf2-example.onnx')
    return tostring(model:input_fact(1))
"#;
let result: String = lua.load(script).eval()?; // returns: "unk__6,100,F32"

Testing

$ make check