mestrevh/computer-vision-beans

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)