Lib.rs

使用Rust的Deep Danbooru

不使用Python的多标签动漫图像分类

快速使用

# let model_path = projects.join("deep-danbooru-models/models/deepdanbooru-2021.onnx");
# let image_path = projects.join("deep-danbooru-models/tests/pixel-105715609.jpg");
predict_by_danbooru2021(&model_path, &image_path)

完整使用

let runtime = Arc::new(Environment::builder().build()?);
let mut model = DeepDanbooru::new(&runtime, model.as_ref())?;
model.set_tags(TAGS2021);
let image = Reader::open(image.as_ref()).unwrap().decode().unwrap();
let result = model.predict(&image).unwrap();