metadata
license: apache-2.0
tags:
- vision
- image-classification
datasets:
- AI-Lab-Makerere/beans
metrics:
- accuracy
model-index:
- name: vit-beans-finetuned
results:
- task:
type: image-classification
name: Image Classification
dataset:
name: Beans
type: AI-Lab-Makerere/beans
metrics:
- name: Accuracy
type: accuracy
value: 0.95
vit-beans-finetuned
Fine-tuned ViT-Base on the Beans dataset.
Usage
from transformers import pipeline
clf = pipeline('image-classification', model='PR0G3T/vit-beans-finetuned')
Training
Trained for 1 epoch, batch 16, lr 5e-5, optimizer AdamW.
Limitations
Model specific to bean-leaf images.