Crate deep_danbooru
source ·Expand description
Deep Danbooru in rust
Multi-labels anime image classification without python.
Fast usage
ⓘ
predict_by_danbooru2021(&model_path, &image_path)
Full usage
ⓘ
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();
Structs
Constants
Functions
- One-time prediction of tags in pictures, suitable for command line usage.