Skip to content

Melochi127/Machine-Learning-image-classification-

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

5 Commits
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๐ŸŒธ Oxford Flowers 102 - Image Classification with CNN and EfficientNetB0

This project implements an end-to-end image classification model using the Oxford Flowers 102 dataset. It begins with a Convolutional Neural Network (CNN) built from scratch, then transitions to using EfficientNetB0, the baseline model from "Rethinking Model Scaling for Convolutional Neural Networks."

The workflow includes data preprocessing, advanced data augmentation, regularization, and performance evaluation with the goal of minimizing overfitting and maximizing classification accuracy.


๐Ÿง  Key Features

  • ๐Ÿ“ฆ Dataset: Oxford 102 Flower Categories
  • ๐Ÿงฑ Model 1: Custom CNN (built from scratch)
  • โšก Model 2: EfficientNetB0 (pre-trained on ImageNet, fine-tuned on Flowers102)
  • ๐Ÿ”„ Augmentation: RandomFlip, Rotation, Zoom, Contrast, etc.
  • ๐Ÿ›ก๏ธ Regularization: Dropout, EarlyStopping, L2 Weight Decay
  • ๐Ÿ“Š Metrics: Accuracy, Loss curves, Confusion Matrix, Top-5 Accuracy

๐Ÿ› ๏ธ Tech Stack

  • Python 3.x
  • TensorFlow / Keras
  • NumPy, Matplotlib, seaborn
  • tensorflow_hub, tensorflow_datasets

๐Ÿ“ Project Structure

About

Academic project for Image Classification

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors