1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
use anyhow::{Context, Result};
use std::fs;
use test_programs::nn::{sort_results, witx};
pub fn main() -> Result<()> {
let graph = witx::load_by_name(
"fixtures",
witx::GraphEncoding::Openvino,
witx::ExecutionTarget::CPU,
)?;
let tensor: Vec<u8> = fs::read("fixture/tensor.bgr")
.context("the tensor file to be mapped to the fixture directory")?;
let results = witx::classify(graph, tensor)?;
let top_five = &sort_results(&results)[..5];
println!("found results, sorted top 5: {top_five:?}");
Ok(())
}
|
