Skip to content

Latest commit

 

History

History
39 lines (28 loc) · 1.69 KB

README.md

File metadata and controls

39 lines (28 loc) · 1.69 KB

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