metadata
license: apache-2.0
tags:
- vision
- image-classification
- fine-tuning
- transformer
- vit
datasets:
- beans
metrics:
- accuracy
- f1
Vision Transformer Fine-tuned on Beans
Este modelo é um Vision Transformer (ViT) fine-tuned no dataset Beans para classificação de imagens.
📝 Descrição do Modelo
O modelo foi treinado no dataset Beans, que contém imagens de folhas de feijão classificadas em 3 categorias:
angular_leaf_spot(mancha angular da folha),bean_rust(ferrugem do feijão),healthy(saudável).
A fine-tuning foi feita a partir do modelo pré-treinado google/vit-base-patch16-224-in21k, ajustado para 3 classes.
📌 Uso Pretendido
Este modelo pode ser usado para tarefas de classificação de imagens, ajudando na identificação de doenças em folhas de feijão.
🚀 Como Usar
Você pode usar o modelo diretamente com o Hugging Face Transformers:
from transformers import pipeline
classifier = pipeline('image-classification', model='your-username/your-model-name')
results = classifier('path/to/your/image.jpg')
print(results)