Image Classification with Transfer Learning

Project Overview

This project implements a 5-class image classifier using the VGG16 pre-trained model with custom dense layers. It leverages transfer learning to optimize performance on a small dataset of images.

Dataset

The dataset used for this project can be accessed from the following link:
Flowers Dataset

Features

• Transfer learning with VGG16 for efficient feature extraction.
• Custom fully connected layers for multi-class classification.
• Data augmentation techniques for improved generalization.
• Optimized with TensorFlow and Keras.

Setup Instructions

To set up this project locally:

1. Clone the repository:

   git clone <repository-url>