This project demonstrates image classification on the CIFAR-10 dataset using transfer learning with the pre-trained VGG16 model. The implementation is done in Google Colab and includes data preprocessing, model adaptation, training, evaluation, and result visualization using TensorFlow and Keras.
To run this project in Google Colab, you don't need to install anything locally. All necessary packages are available in the Colab environment.
The project consists of the following steps:
- Setup and Data Preparation: Load and preprocess the CIFAR-10 dataset.
- Load and Adapt Pre-trained Model: Use the VGG16 model pre-trained on ImageNet and adapt it for CIFAR-10 classification.
- Train the Model: Train the adapted model using the training data.
- Evaluate the Model: Evaluate the model's performance on the test data.
- Visualize Results: Plot training and validation accuracy and loss.