Update README.md
Browse files
README.md
CHANGED
|
@@ -1,25 +1,92 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Synapse-Image-Classifier-Nano
|
| 2 |
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
-
|
| 6 |
-
-
|
| 7 |
-
-
|
| 8 |
-
-
|
| 9 |
-
- **Acurácia Final:** 64.62% (após 20 épocas)
|
| 10 |
-
- **Desenvolvedor:** Comunidade Synapse BR
|
| 11 |
|
| 12 |
-
##
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
-
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
-
## Como usar
|
| 19 |
```python
|
| 20 |
import torch
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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.
|