Lib.rs

Rust 对 Caffe 深度学习库的 FFI 包装。

需要使用 ajtulloch/caffe:caffe-ffi (https://github.com/ajtulloch/caffe/tree/caffe-ffi) 构建的 caffe 来通过 FFI 暴露必要的结构。

let mut net = caffe::Net::new(Path::new("test-data/lenet.prototxt"),
                              caffe::Phase::Test);
net.copy_trained_layers_from(Path::new("test-data/lenet.caffemodel"));
let mut data_blob = net.blob("data");
let mut ones: Vec<_> = repeat(1.0 as f32)
                       .take(data_blob.len())
                       .collect();
data_blob.set_data(ones.as_mut_slice());
net.forward_prefilled();
let prob_blob = net.blob("prob");
let probs = prob_blob.as_slice();
println!("{:?}", probs.to_vec());
assert_eq!(probs[0], 0.06494621)