tract-onnx 0.5.3

Tiny, no-nonsense, self contained, TensorFlow and ONNX inference
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use std::io::{BufRead, Write};
use std::{env, fs, io, path};

fn main() {
    let workdir = path::PathBuf::from(env::var("OUT_DIR").unwrap()).join("protobuf-generated");
    dbg!(&workdir);
    let _ = fs::create_dir_all(&workdir);
    protobuf_codegen_pure::run(protobuf_codegen_pure::Args {
        out_dir: workdir.to_str().unwrap(),
        input: &["protos/onnx/onnx.proto"],
        includes: &["protos"],
        customize: protobuf_codegen_pure::Customize { ..Default::default() },
    })
    .expect("protoc");

    let mut broken = workdir.join("onnx.rs");
    let mut fixed = broken.clone();
    fixed.set_extension("rs");
    broken.set_extension("rs.orig");
    fs::rename(&fixed, &broken).unwrap();
    let mut f = fs::File::create(fixed).unwrap();
    for line in io::BufReader::new(fs::File::open(broken).unwrap()).lines() {
        let line = line.unwrap();
        if !line.starts_with("#![") && !line.starts_with("//!") {
            writeln!(f, "{}", line).unwrap();
        }
    }
}