| # WD 1.4 ViT Tagger V2 | |
| Supports ratings, characters and general tags. | |
| Trained using https://github.com/SmilingWolf/SW-CV-ModelZoo. | |
| TPUs used for training kindly provided by the [TRC program](https://sites.research.google/trc/about/). | |
| ## Dataset | |
| Last image id: 5944504 | |
| Trained on Danbooru images with IDs modulo 0000-0899. | |
| Validated on images with IDs modulo 0950-0999. | |
| Images with less than 10 general tags were filtered out. | |
| Tags with less than 600 images were filtered out. | |
| ## Validation results | |
| `P=R: threshold = 0.3537, F1: 0.6770` | |
| ## Final words | |
| Subject to change and updates. | |
| Downstream users are encouraged to use tagged releases rather than relying on the head of the repo. | |