CNN-based image classification on CIFAR-10 dataset using TensorFlow/Keras. The model classifies images into 10 categories such as airplane, car, dog, and cat. Includes data preprocessing, training, and evaluation with accuracy visualization.
This project implements an image classification model using Convolutional Neural Networks (CNN) on the CIFAR-10 dataset.
To classify images into 10 categories such as airplane, car, bird, cat, dog, automobile, etc.....
- Python
- TensorFlow / Keras
- Google Colab
CIFAR-10 dataset (built into TensorFlow)
Achieved ~70% accuracy using a simple CNN model.
- Add data augmentation,
- Use transfer learning (ResNet, MobileNet),
- Hyperparameter tuning
Open the notebook in Google Colab and run all cells.