mestrevh commited on
Commit
5bfbd28
·
1 Parent(s): bbebd6c

Add the file

Browse files
Files changed (1) hide show
  1. README.md +20 -9
README.md CHANGED
@@ -7,27 +7,38 @@ tags:
7
  - transformer
8
  - vit
9
  datasets:
10
- - cifar10
11
- - imagenet
12
  metrics:
13
  - accuracy
14
  - f1
15
  ---
16
 
17
- # Vision Transformer Fine-tuned on CIFAR-10
18
 
19
- This model is a Vision Transformer (ViT) fine-tuned on the CIFAR-10 dataset for image classification.
20
 
21
- ## Model Description
22
- The model was trained on the CIFAR-10 dataset, which consists of 60,000 32x32 color images in 10 classes, with 6,000 images per class. The fine-tuning was performed on the pretrained ViT base model.
23
 
24
- ## Intended Use
25
- This model is intended for image classification tasks.
 
 
 
 
 
 
 
 
 
 
 
 
 
26
 
27
- ## How to Use
28
  ```python
29
  from transformers import pipeline
30
 
31
  classifier = pipeline('image-classification', model='your-username/your-model-name')
32
  results = classifier('path/to/your/image.jpg')
33
  print(results)
 
 
7
  - transformer
8
  - vit
9
  datasets:
10
+ - beans
 
11
  metrics:
12
  - accuracy
13
  - f1
14
  ---
15
 
16
+ # Vision Transformer Fine-tuned on Beans
17
 
18
+ Este modelo é um **Vision Transformer (ViT)** fine-tuned no dataset **Beans** para classificação de imagens.
19
 
20
+ ## 📝 Descrição do Modelo
 
21
 
22
+ O modelo foi treinado no dataset **Beans**, que contém imagens de folhas de feijão classificadas em **3 categorias**:
23
+
24
+ - `angular_leaf_spot` (mancha angular da folha),
25
+ - `bean_rust` (ferrugem do feijão),
26
+ - `healthy` (saudável).
27
+
28
+ A fine-tuning foi feita a partir do modelo pré-treinado `google/vit-base-patch16-224-in21k`, ajustado para 3 classes.
29
+
30
+ ## 📌 Uso Pretendido
31
+
32
+ Este modelo pode ser usado para tarefas de **classificação de imagens**, ajudando na identificação de doenças em folhas de feijão.
33
+
34
+ ## 🚀 Como Usar
35
+
36
+ Você pode usar o modelo diretamente com o **Hugging Face Transformers**:
37
 
 
38
  ```python
39
  from transformers import pipeline
40
 
41
  classifier = pipeline('image-classification', model='your-username/your-model-name')
42
  results = classifier('path/to/your/image.jpg')
43
  print(results)
44
+ ```