mlua-tract

Lua 绑定 tract，tract 是一个小型、简单、自包含的 Rust 基于的 Tensorflow 和 ONNX 推理运行时。

安装

使用 MLua 的功能之一将以下内容添加到您的 Rust 项目中：[lua51, lua52, lua53, lua54, luajit, luajit52]。

$ cargo add mlua-tract --features luajit

使用

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"

测试

$ make check