Skip to content

mosmeh/witchbooru

Repository files navigation

Witchbooru

build

Who is this character?

Web demo

Witchbooru recognizes anime characters in images.

It combines an accurate deep neural network and a lightweight naive Bayes classifier, enabling low-cost updates of the model to handle the ever-increasing number of characters.

Training / preparing model

See scripts/README

Lambda function

Build

cargo build -p witchbooru-lambda --release --target x86_64-unknown-linux-musl
zip -j lambda.zip ./target/x86_64-unknown-linux-musl/release/bootstrap

Test locally

docker compose up -d
aws s3 mb --endpoint=http://localhost:4566 s3://witchbooru-model/
aws s3 cp --endpoint=http://localhost:4566 --recursive ./model/ s3://witchbooru-model/
sam local start-api --docker-network localstack

Deploy to AWS

sam deploy --guided

# Upload model files to S3 as appropriate

Command-line interface

cargo run -p witchbooru-cli --release -- /path/to/img -m ./model

Frontend

See frontend/README

About

Image recognition for anime characters

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published