Um projeto de classificação de imagens utilizando deep learning com TensorFlow e Keras. O objetivo é treinar um modelo de rede neural convolucional (CNN) para reconhecer diferentes classes de imagens.
- Dataset utilizado: The CIFAR-10 dataset
- ré-processamento e augmentação de imagens
- Treinamento de modelo CNN com Keras
- Visualização de acurácia e perda
- Avaliação em conjunto de validação
- Salvamento do modelo treinado
- Python 3
- TensorFlow
- Matplotlib
- NumPy
- Jupyter Notebook
-
Clone este repositório:
git clone https://github.com/MatheusFL99/image_classifier.git cd image_classifier -
Crie um ambiente virtual python:
python -m venv venv source venv/bin/activate # Linux/macOS venv\Scripts\activate # Windows
-
Instale as dependências:
pip install -r requirements.txt
-
Execute o notebook
image-classifier.ipynb -
Para testar a classificação de uma imagem
- Substitua o arquivo
data/image.jpgpela imagem que deseja - Rode as células da parte do Deploy no notebook.
- Substitua o arquivo