Downloads · 30 days
7
27% of all-time downloads
wd-v1-4-convnextv2-tagger-v2
PUXIUBAI/wd-v1-4-convnextv2-tagger-v2
wd-v1-4-convnextv2-tagger-v2 is a machine learning model from PUXIUBAI. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for tf-keras. The card lists the license as apache-2.0.
Supports ratings, characters and general tags.
All-time downloads
26
Public
Repo size
783 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.onnx388 MB · 50%
At a glance
- Library
- tf-keras
- License
- apache-2.0
- Access
- Public
- Created
- Jul 11, 2026
- Updated
- Jul 11, 2026
- SHA
- 549feb7e
Model card
From the Hugging Face model README
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.
- Library
- tf-keras
- License
- apache-2.0
- Created
- Jul 11, 2026
- Updated
- Jul 11, 2026