Olá a todos, este projeto serve para meu estudo continuo e para ajudar a todos a treinar o conhecimento.
Este projeto demonstra a construção e o treinamento de uma Convolutional Neural Network (CNN) para classificação de imagens utilizando o dataset CIFAR-10.
O objetivo é desenvolver um classificador de imagens capaz de identificar 10 classes diferentes de objetos presentes no dataset CIFAR-10. O projeto inclui todas as etapas desde o carregamento e visualização do dataset até a construção, treinamento e avaliação do modelo.
- PyTorch
- torchvision
- matplotlib
- numpy
- scikit-learn
- Clone o repositório:
git clone https://github.com/airtonmega/image-classifier-cifar10.git cd image-classifier-cifar10