Sponsored by
AdsCreator.com
p1atdev / wd-swinv2-tagger-v3-hf
Model's Last Updated: March 11 2024
image-classification
Introduction of wd-swinv2-tagger-v3-hf
Model Details of wd-swinv2-tagger-v3-hf
WD SwinV2 Tagger v3 with 🤗 transformers
Converted from SmilingWolf/wd-swinv2-tagger-v3 to transformers library format.
Example
Installation
pip install transformers
Pipeline
from transformers import pipeline
pipe = pipeline(
"image-classification",
model="p1atdev/wd-swinv2-tagger-v3-hf",
trust_remote_code=True,
)
print(pipe("sample.webp", top_k=15))
#[{'label': '1girl', 'score': 0.9973934888839722},
# {'label': 'solo', 'score': 0.9719744324684143},
# {'label': 'dress', 'score': 0.9539461135864258},
# {'label': 'hat', 'score': 0.9511678218841553},
# {'label': 'outdoors', 'score': 0.9438753128051758},
# ...
AutoModel
from PIL import Image
import numpy as np
import torch
from transformers import (
AutoImageProcessor,
AutoModelForImageClassification,
)
MODEL_NAME = "p1atdev/wd-swinv2-tagger-v3-hf"
model = AutoModelForImageClassification.from_pretrained(
MODEL_NAME,
)
processor = AutoImageProcessor.from_pretrained(MODEL_NAME, trust_remote_code=True)
image = Image.open("sample.webp")
inputs = processor.preprocess(image, return_tensors="pt")
with torch.no_grad():
outputs = model(**inputs.to(model.device, model.dtype))
logits = torch.sigmoid(outputs.logits[0]) # take the first logits
# get probabilities
results = {model.config.id2label[i]: logit.float() for i, logit in enumerate(logits)}
results = {
k: v for k, v in sorted(results.items(), key=lambda item: item[1], reverse=True) if v > 0.35 # 35% threshold
}
print(results) # rating tags and character tags are also included
#{'1girl': tensor(0.9974),
# 'solo': tensor(0.9720),
# 'dress': tensor(0.9539),
# 'hat': tensor(0.9512),
# 'outdoors': tensor(0.9439),
# ...
Accelerate with 🤗 Optimum
Maybe about 30% faster and about 50% light weight model size than transformers version, but the accuracy is slightly degraded.
pip install optimum[onnxruntime]
-from transformers import pipeline
+from optimum.pipelines import pipeline
pipe = pipeline(
"image-classification",
model="p1atdev/wd-swinv2-tagger-v3-hf",
trust_remote_code=True,
)
print(pipe("sample.webp", top_k=15))
#[{'label': '1girl', 'score': 0.9966088533401489},
# {'label': 'solo', 'score': 0.9740601778030396},
# {'label': 'dress', 'score': 0.9618403911590576},
# {'label': 'hat', 'score': 0.9563733339309692},
# {'label': 'outdoors', 'score': 0.945336639881134},
# ...
Labels
All of rating tags have prefix
rating:
and character tags have prefix
character:
.
-
Rating tags:
rating:general,rating:sensitive, ... -
Character tags:
character:frieren,character:hatsune miku, ...
Runs of p1atdev wd-swinv2-tagger-v3-hf on huggingface.co
2.2K
Total runs
0
24-hour runs
0
3-day runs
0
7-day runs
0
30-day runs
More Information About wd-swinv2-tagger-v3-hf huggingface.co Model
Url of wd-swinv2-tagger-v3-hf
wd-swinv2-tagger-v3-hf huggingface.co Url
Provider of wd-swinv2-tagger-v3-hf huggingface.co
Other API from p1atdev
Total runs:
1.5K
Run Growth:
-1.7K
Growth Rate:
-113.05%
Total runs:
303
Run Growth:
154
Growth Rate:
50.66%
Total runs:
132
Run Growth:
-63
Growth Rate:
-47.37%
Total runs:
104
Run Growth:
0
Growth Rate:
0.00%
Total runs:
103
Run Growth:
4
Growth Rate:
3.88%
Total runs:
54
Run Growth:
45
Growth Rate:
83.33%
Total runs:
43
Run Growth:
31
Growth Rate:
72.09%
Total runs:
27
Run Growth:
-6
Growth Rate:
-22.22%
Total runs:
22
Run Growth:
9
Growth Rate:
40.91%
Total runs:
19
Run Growth:
8
Growth Rate:
38.10%
Total runs:
18
Run Growth:
-17
Growth Rate:
-94.44%
Total runs:
13
Run Growth:
10
Growth Rate:
76.92%
Total runs:
10
Run Growth:
1
Growth Rate:
10.00%
Total runs:
10
Run Growth:
4
Growth Rate:
40.00%
Total runs:
8
Run Growth:
-1
Growth Rate:
-12.50%
Total runs:
5
Run Growth:
2
Growth Rate:
40.00%
Total runs:
3
Run Growth:
-7
Growth Rate:
-233.33%
Total runs:
2
Run Growth:
0
Growth Rate:
0.00%
Total runs:
2
Run Growth:
-1
Growth Rate:
-50.00%
Total runs:
2
Run Growth:
-3
Growth Rate:
-150.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
-4
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
-3
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%