tensorflow-serving

This package integrates Tensorflow Serving with Rust.

License: Apache-2.0

lib.rs:

This package integrates Tensorflow Serving with Rust.

let serving = TensorflowServing::new()
.hostname("127.0.0.1")
.port(8500)
.build()?;

// Your test image is stored at `img_path`, or it is an already open `image::DynamicImage`
let prediction = serving.predict(img_path, "resnet")?;

println!("Probabilities: {:?}", prediction.probabilities);
println!("Top index: {}", prediction.max_idx);