Run SmilingWolfs wd-convnext-tagger-v3 model on mobile!
The app is currently in closed beta and needs testers! You can join the testing group here to gain access.
Due to new Google Play policies, 20 testers are needed for 14 days before the app can go public.
Android is currently supported!
I think iOS could easily be supported but I haven't tested it. Same goes for web.
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.
If you get bad results, it's highly likely that I did something wrong with the preprocessing. Any tips for improvement would be appreciated.
PRs are welcome!
- Download and install the latest dev build from releases
- Clone repo
- Run
yarn - Download model and place it in assets/models
- Run
yarn start
- This app uses expo-share-intent and will need additional setup for iOS.