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]}')

Downloads last month: 6

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Examples

Image Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.