WD 1.4 ConvNextV2 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 .

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.3710, F1 = 0.6862

Final words

Subject to change and updates. Downstream users are encouraged to use tagged releases rather than relying on the head of the repo.