Skip to content

Caltech-101 image classification using EfficientNet-B4 with 98.56% training accuracy and 93% validation accuracy.

License

Notifications You must be signed in to change notification settings

SyedaEmanSaleem/Caltech101-project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

6 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

Caltech-101 Image Classification

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%


๐Ÿ“‚ Contents

  • Caltech101_EfficientNet.ipynb โ€“ Main notebook with model implementation and training
  • requirements.txt โ€“ Dependencies used in the project
  • training_log.csv โ€“ Training curves (accuracy/loss per epoch)

โš™๏ธ Requirements

  • Python 3.x
  • TensorFlow / Keras
  • NumPy
  • Matplotlib
  • scikit-learn

Install dependencies:

pip install -r requirements.txt

๐Ÿš€ How to Run

  1. Download the Caltech-101 dataset from Caltech Dataset Page or via TensorFlow Datasets.
  2. Place the dataset in a folder named data/ (or update the dataset path in the notebook).
  3. Run the notebook:
jupyter notebook Caltech101_EfficientNet.ipynb

๐Ÿท๏ธ Topics

cnn image-classification deep-learning transfer-learning efficientnet caltech101


About

Caltech-101 image classification using EfficientNet-B4 with 98.56% training accuracy and 93% validation accuracy.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published