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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/*!
# Cervo
Cervo is a toolkit we use for ML integration in games. The core
use-case is for RL-agents, and some utilities for managing brain assets.
## Cervo Core
The core crate focuses on wrappers for tract models. It adds a few
different modes for running inferers, as well as data injectors for
stochastic policies (e.g. SAC).
```no_run
# fn load_bytes(s: &str) -> std::io::Cursor<Vec<u8>> { std::io::Cursor::new(vec![]) }
# use cervo_onnx::tract_onnx;
# use cervo_onnx::tract_onnx::prelude::*;
use cervo_core::prelude::{BasicInferer, InfererExt};
let mut model_data = load_bytes("model.onnx");
let inference_model = tract_onnx::onnx().model_for_read(&mut model_data)?;
let inferer = BasicInferer::from_model(inference_model)?
.with_default_epsilon("noise");
# Ok::<(), Box<dyn std::error::Error>>(())
```
## Cervo Asset
To support isomg NNEF and ONNX interchangeably we have a small
wrapping binary format which can contain either type of data, helping
keep track of which data is what.
```no_run
# fn load_bytes(s: &str) -> Vec<u8> { vec![] }
use cervo_asset::{AssetData, AssetKind};
let model_data = load_bytes("model.onnx");
let asset = AssetData::new(AssetKind::Onnx, model_data);
let nnef_asset = asset.to_nnef(None)?; // convert to a symbolic NNEF asset
let inferer = asset.load_basic();
let nnef_inferer = nnef_asset.load_fixed(&[42]);
# Ok::<(), Box<dyn std::error::Error>>(())
```
## Cervo ONNX and Cervo NNEF
These are simple intermediates helping Cervo Asset, but can also be used directly.
```no_run
# fn load_bytes(s: &str) -> std::io::Cursor<Vec<u8>> { std::io::Cursor::new(vec![]) }
use cervo_core::prelude::InfererExt;
let model_data = load_bytes("model.onnx");
let model = cervo_onnx::builder(model_data)
.build_memoizing(&[])?
.with_default_epsilon("epsilon");
# Ok::<(), Box<dyn std::error::Error>>(())
```
*/
pub use cervo_asset as asset;
pub use cervo_core as core;
pub use cervo_nnef as nnef;
pub use cervo_onnx as onnx;