ViT Beans Classifier

Fine-tuned google/vit-base-patch16-224-in21k on the official beans dataset for 4 epochs. It classifies bean leaves as angular_leaf_spot, bean_rust, or healthy.

Test result

Metric Value
Accuracy 0.9609

Usage

from transformers import pipeline
classifier = pipeline("image-classification", model="Wadden/vit-beans-demo")
classifier("leaf.jpg")

Interactive demo

The quantized ONNX export is executed directly in the browser with Transformers.js:

Open the interactive Space

Downloads last month
84
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train Wadden/vit-beans-demo

Space using Wadden/vit-beans-demo 1