Ir para o conteúdo
ModelListadosExterno
Baixar

Cifar Cnn

A 3-block CNN trained from scratch on CIFAR-10 using PyTorch.

TipoOtherVersãoRevision f2bb2f9Cnota

Fonte do modelo

Tipo
Other
Tamanho do arquivo
5,13 MB
Versão
vRevision f2bb2f9
Atualizado
11 de out. de 2026
Downloads
5
Fonte
Hugging Face

Descrição da fonte

CIFAR-10 CNN Classifier

A 3-block CNN trained from scratch on CIFAR-10 using PyTorch.

Classes

plane, car, bird, cat, deer, dog, frog, horse, ship, truck

Training

  • Architecture: 3× Conv blocks (32→64→128 channels) + FC classifier
  • Optimizer: Adam + OneCycleLR scheduler
  • Augmentation: RandomCrop, RandomFlip, ColorJitter, RandomErasing
  • Validation accuracy: ~80%+ (after 64 epochs)

Usage

from transformers import pipeline
from PIL import Image

classifier = pipeline(
    "image-classification",
    model="MarkivDunhar/cifar-cnn"
)

result = classifier("your_image.jpg")
print(result)
# [{'label': 'dog', 'score': 0.91}]

Fontes

1 fonte
Hugging FacePrimárioVersão f2bb2f940fdbf29323c47facc271ab168d3f5e17
huggingface.co/MarkivDunhar/cifar-cnn/tree/f2bb2f940fdbf29323c47facc271ab168d3f5e17

Verificado 26 de ago.

Artefatos de modelo

1 artefato

model.safetensors

safetensors · 5,13 MB · SHA-256 6c84dcd8c997…0a06 · Hugging Face

Baixar

Trechos de fonte

2 trechos

--- license: apache-2.0 tags: - image-classification - pytorch - cifar10 datasets: - uoft-cs/cifar10 --- # CIFAR-10 CNN Classifier A 3-block CNN trained from scratch on CIFAR-10 using PyTorch. ## Classes plane, car, bird, cat, deer, dog, frog, horse, ship, truck ## Training - Architecture: 3× Conv blocks (32→64→128 channels) + FC classifier - Optimizer: Adam + OneCycleLR scheduler - Augmentation: RandomCrop, RandomFlip, ColorJitter, RandomErasing - Validation accuracy: ~80%+ (after 64 epochs) ## Usage ```python from transformers import pipeline from PIL import Image classifier = pipeline( "image-classification", model="MarkivDunhar/cifar-cnn" ) result = classifier("your_image.jpg") print(result) # [{'label': 'dog', 'score': 0.91}] ```

Source context: 5 downloads · 0 likes · Pipeline image-classification · Repo MarkivDunhar/cifar-cnn