|
|
--- |
|
|
tags: |
|
|
- vision |
|
|
- image-classification |
|
|
- transformers |
|
|
datasets: |
|
|
- beans |
|
|
license: apache-2.0 |
|
|
--- |
|
|
|
|
|
# 🏆 Modelo de Classificação de Imagens - Vision Transformer (ViT) |
|
|
|
|
|
Este modelo foi treinado para **classificação de imagens** usando o **Vision Transformer (ViT)**. |
|
|
|
|
|
📌 **Repositório do Space (Aplicação Web)**: |
|
|
👉 [mestrevh/image-classification-vit](https://huggingface.co/spaces/mestrevh/image-classification-vit) |
|
|
|
|
|
--- |
|
|
|
|
|
## 📌 **Descrição do Modelo** |
|
|
Este modelo foi treinado para classificar imagens em diferentes categorias com base no dataset **Beans**. |
|
|
|
|
|
🔹 **Modelo Base**: [`google/vit-base-patch16-224-in21k`](https://huggingface.co/google/vit-base-patch16-224-in21k) |
|
|
🔹 **Dataset utilizado**: [`beans`](https://huggingface.co/datasets/beans) |
|
|
🔹 **Tarefa**: Classificação de imagens |
|
|
|
|
|
--- |
|
|
|
|
|
## 🚀 **Como Usar** |
|
|
Você pode carregar este modelo diretamente com `transformers`: |
|
|
|
|
|
```python |
|
|
from transformers import pipeline |
|
|
|
|
|
# Carregar modelo treinado do Hugging Face Hub |
|
|
model_id = "mestrevh/NOME_DO_MODELO" |
|
|
classifier = pipeline("image-classification", model=model_id) |
|
|
|
|
|
# Fazer uma predição |
|
|
image_path = "caminho/para/sua/imagem.jpg" |
|
|
result = classifier(image_path) |
|
|
print(result) |
|
|
|
