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.ipynbcnn image-classification deep-learning transfer-learning efficientnet caltech101