How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("image-classification", model="t7hom/vit-beans-classifier")
pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification

processor = AutoImageProcessor.from_pretrained("t7hom/vit-beans-classifier")
model = AutoModelForImageClassification.from_pretrained("t7hom/vit-beans-classifier", device_map="auto")
Quick Links

Vision Transformer for Bean Leaf Disease Classification

  • Authors: s_102413_109626
  • Backbone: WinKawaks/vit-tiny-patch16-224
  • Dataset: beans
  • Classes: angular_leaf_spot, bean_rust, healthy

Test Metrics

  • Test Accuracy: 0.9609
  • Test F1 Macro: 0.9615
Downloads last month
6
Safetensors
Model size
5.52M params
Tensor type
F32
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Dataset used to train t7hom/vit-beans-classifier

Space using t7hom/vit-beans-classifier 1