Downloads · 30 days
11
2% of all-time downloads
lrzjason/hand-classifier
hand-classifier is a image classification model from lrzjason. 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.
a hand classifier detect the image is showing left hand or right hand
Downloads · 30 days
11
2% of all-time downloads
All-time downloads
554
Public
Parameters
85.8M
343 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors343 MB · 100%
From the Hugging Face model README
a hand classifier detect the image is showing left hand or right hand
it is trained with hagrid dataset on vit model, google/vit-base-patch16-224-in21k.
Limitation: it couldn't predict both hand and no hand images.
inputs = image_processor(image, return_tensors="pt")
with torch.no_grad():
logits = model(**inputs).logits
predicted_label = logits.argmax(-1).item()
print(f'predicted_label: {model.config.id2label[predicted_label]}')