Skip to content

Multi-labels anime image classification in rust

License

Notifications You must be signed in to change notification settings

libwaifu/deep-danbooru

Repository files navigation

Deep Danbooru for rust

Multi-labels anime image classification without python.

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();

Develop

  • Clone with models
# https
git clone --recursive https://github.com/libwaifu/deep-danbooru.git
# git
git clone --recursive git@github.com:libwaifu/deep-danbooru.git

or download manually: deep-danbooru/models

  • Pull latest models
git pull && git submodule update --remote

About

Multi-labels anime image classification in rust

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages