Rafaelvieira432 commited on
Commit
f576da8
·
verified ·
1 Parent(s): 295685d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +84 -17
README.md CHANGED
@@ -1,25 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Synapse-Image-Classifier-Nano
2
 
3
- Este modelo é uma implementação customizada da arquitetura **ResNet-18 (Nano)** treinada no dataset **CIFAR-100**.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
 
5
- ## Detalhes do Modelo
6
- - **Arquitetura:** ResNet-18 Nano (Thin version)
7
- - **Parâmetros:** 4,397,580
8
- - **Dataset:** CIFAR-100
9
- - **Acurácia Final:** 64.62% (após 20 épocas)
10
- - **Desenvolvedor:** Comunidade Synapse BR
11
 
12
- ## Hiperparâmetros de Treino
13
- - **Otimizador:** Adam
14
- - **Learning Rate:** 0.001
15
- - **Batch Size:** 128
16
- - **Augmentation:** RandomHorizontalFlip, RandomCrop(32, padding=4)
 
 
 
17
 
18
- ## Como usar
19
  ```python
20
  import torch
21
- # Certifique-se de ter a definição da classe ResNetNano carregada
22
- model = ResNetNano(BasicBlock, [2, 2, 2, 2], num_classes=100, width=40)
23
- model.load_state_dict(torch.load('pytorch_model.bin'))
 
 
 
 
 
 
 
 
24
  model.eval()
25
- ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - uoft-cs/cifar100
4
+ language:
5
+ - pt
6
+ license: apache-2.0
7
+ pipeline_tag: image-classification
8
+ tags:
9
+ - computer-vision
10
+ - image-classification
11
+ - resnet
12
+ - cifar100
13
+ - pytorch
14
+ - synapse-br
15
+ ---
16
+
17
  # Synapse-Image-Classifier-Nano
18
 
19
+ **Synapse-Image-Classifier-Nano** é um classificador de imagens leve desenvolvido pela **Comunidade Synapse BR**.
20
+
21
+ O modelo utiliza uma implementação personalizada da arquitetura **Thin ResNet-18**, projetada para reduzir o número de parâmetros mantendo um bom desempenho em tarefas de classificação de imagens.
22
+
23
+ ## Informações do Modelo
24
+
25
+ | Propriedade | Valor |
26
+ |------------|-------|
27
+ | Arquitetura | Thin ResNet-18 |
28
+ | Parâmetros | 4,397,580 |
29
+ | Framework | PyTorch |
30
+ | Dataset | CIFAR-100 |
31
+ | Classes | 100 |
32
+ | Entrada | RGB 32×32 |
33
+ | Épocas | 20 |
34
+ | Batch Size | 128 |
35
+ | Otimizador | Adam |
36
+ | Learning Rate | 0.001 |
37
+ | Top-1 Accuracy | **64.62%** (CIFAR-100 Test Set) |
38
+
39
+ ## Dataset
40
+
41
+ O modelo foi treinado utilizando o dataset **CIFAR-100**, composto por:
42
 
43
+ - 50.000 imagens para treinamento
44
+ - 10.000 imagens para teste
45
+ - 100 classes
46
+ - Imagens RGB de 32×32 pixels
 
 
47
 
48
+ ## Data Augmentation
49
+
50
+ Durante o treinamento foram utilizadas as seguintes técnicas:
51
+
52
+ - RandomHorizontalFlip
53
+ - RandomCrop(32, padding=4)
54
+
55
+ ## Como utilizar
56
 
 
57
  ```python
58
  import torch
59
+
60
+ # Certifique-se de possuir a implementação da arquitetura Thin ResNet-18
61
+
62
+ model = ResNetNano(
63
+ BasicBlock,
64
+ [2, 2, 2, 2],
65
+ num_classes=100,
66
+ width=40
67
+ )
68
+
69
+ model.load_state_dict(torch.load("pytorch_model.bin"))
70
  model.eval()
71
+
72
+ Resultados
73
+
74
+ Métrica| Valor
75
+ Top-1 Accuracy| 64.62%
76
+
77
+ Limitações
78
+
79
+ - Treinado exclusivamente no dataset CIFAR-100.
80
+ - Espera imagens RGB de resolução 32×32.
81
+ - Não foi treinado para classificação de imagens de alta resolução.
82
+ - O desempenho pode diminuir significativamente em domínios diferentes do CIFAR-100.
83
+
84
+ Objetivo
85
+
86
+ O Synapse-Image-Classifier-Nano foi desenvolvido como um modelo leve para pesquisas, aprendizado e aplicações de classificação de imagens que exigem baixo consumo de recursos computacionais.
87
+
88
+ Desenvolvedor
89
+
90
+ Comunidade Synapse BR
91
+
92
+ Projeto open source focado no desenvolvimento de modelos leves, datasets e ferramentas de inteligência artificial.