Skip to content

A utility app that tags your images with danbooru tags - fully offline

Notifications You must be signed in to change notification settings

KuzuLabz/WaifuTagger

Repository files navigation

WaifuTagger

banner

Hugging Face Hugging Face

Run SmilingWolfs wd-convnext-tagger-v3 model on mobile!

🎉 Google Play Store

Google has proven they hate indie devs, so I won't be uploading anything there. I am considering F-Droid though.

🎮 Platforms

Android is currently supported!

I think iOS could easily be supported but I haven't tested it. Same goes for web.

🤗 Model

To run the model on mobile, I quantized it and added preprocessing using onnxruntime-extentions. This model is available on my HuggingFace. You can also find more details about the process there.

🎯 Model Accuracy

If you get bad results, it's highly likely that I did something wrong with the preprocessing. Any tips for improvement would be appreciated.

Contributing

PRs are welcome!

Development Setup

  1. Download and install the latest dev build from releases
  2. Clone repo
  3. Run yarn
  4. Download model and place it in assets/models
  5. Run yarn start

iOS Notes