This project applies Convolutional Neural Networks (CNNs) with transfer learning (EfficientNet-B6) to classify images from the Caltech-101 dataset. The dataset contains 101 object categories (animals, vehicles, instruments, etc.), making it a challenging image classification task.
โ Training Accuracy: 98.56% โ Validation Accuracy: 93.00%
Caltech101_EfficientNet.ipynb
โ Main notebook with model implementation and trainingrequirements.txt
โ Dependencies used in the projecttraining_log.csv
โ Training curves (accuracy/loss per epoch)
- Python 3.x
- TensorFlow / Keras
- NumPy
- Matplotlib
- scikit-learn
Install dependencies:
pip install -r requirements.txt
- Download the Caltech-101 dataset from Caltech Dataset Page or via TensorFlow Datasets.
- Place the dataset in a folder named
data/
(or update the dataset path in the notebook). - Run the notebook:
jupyter notebook Caltech101_EfficientNet.ipynb
cnn
image-classification
deep-learning
transfer-learning
efficientnet
caltech101