1 个不稳定版本
0.1.0 | 2019年5月17日 |
---|
#749 在 机器学习
1.5MB
29K SLoC
tensorflow-serving
此包将 Tensorflow Serving 与 Rust 集成。
许可证:Apache-2.0
lib.rs
:
此包将 Tensorflow Serving 与 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);
依赖关系
~41MB
~652K SLoC