Downloads · 30 days
8
1% of all-time downloads
umamusume-classifier
furusu/umamusume-classifier
umamusume-classifier is a image classification model from furusu. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
finetuned from https://huggingface.co/google/vit-base-patch16-224-in21k
All-time downloads
934
Public
Repo size
687 MB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin344 MB · 100%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Jan 13, 2023
- Updated
- Jan 13, 2023
- SHA
- e089cbce
Model card
From the Hugging Face model README
finetuned from https://huggingface.co/google/vit-base-patch16-224-in21k
dataset:26k images (train:21k valid:5k)
accuracy of validation dataset is 95%
from transformers import ViTFeatureExtractor, ViTForImageClassification
from PIL import Image
path = 'image_path'
image = Image.open(path)
feature_extractor = ViTFeatureExtractor.from_pretrained('furusu/umamusume-classifier')
model = ViTForImageClassification.from_pretrained('furusu/umamusume-classifier')
inputs = feature_extractor(images=image, return_tensors="pt")
outputs = model(**inputs)
predicted_class_idx = outputs.logits.argmax(-1).item()
print("Predicted class:", model.config.id2label[predicted_class_idx])
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Jan 13, 2023
- Updated
- Jan 13, 2023