Skip to content

πŸ“Œ **15-Class Image Classifier** using EfficientNetB0 & TensorFlow/Keras A transfer learning model for multi-class image classification with data preprocessing, visualization & evaluation. πŸ”§ **Tech Stack**: - EfficientNetB0 (pretrained) - TensorFlow / Keras #ComputerVision #DeepLearning #TransferLearning

License

Notifications You must be signed in to change notification settings

AliRezaKhatibi/15-Class-CNN-Classifier

Repository files navigation

πŸ€– Human Action Recognition System

Advanced 15-class action classification using EfficientNetB0

Python TensorFlow Keras Accuracy License

This project implements a 15-class human action recognition system using transfer learning with EfficientNetB0. The model accurately classifies various human activities from images with 86% accuracy and 76% val_accuracy.

πŸ“¦ Data Statistics

  • Total images: 12,600
  • Image dimensions: 224Γ—224Γ—3
  • Training samples: 10,080 (80%)
  • Test samples: 2,520 (20%)

🏷️ Activity Classes

  • sitting
  • using_laptop
  • hugging
  • sleeping
  • drinking
  • clapping
  • dancing
  • cycling
  • calling
  • laughing
  • eating
  • fighting
  • listening_to_music
  • running
  • texting

🧠 Neural Network Structure

EfficientNetB0 (base model)
β”œβ”€ GlobalAveragePooling2D()
β”œβ”€ BatchNormalization()
β”œβ”€ Dense(128, activation='relu', L2 regularization)
β”œβ”€ Dropout(0.5)
β”œβ”€ Dense(128, activation='relu')
β”œβ”€ Dropout(0.3)
└─ Dense(15, activation='softmax')

βš™οΈ Training Configuration

  • Optimizer: Adam (lr=1e-4)
  • Loss: Categorical Crossentropy
  • Batch Size: 32
  • Base Model Frozen (except BatchNorm layers)

πŸ“₯ Installation

  1. Clone the repository:
    git clone https://github.com/your-username/15-Class-CNN-Classifier.git
  2. Install dependencies:
    pip install -r requirements.txt
  3. Download dataset from Kaggle and place in Data/ directory

πŸ‹οΈ Training the Model

python train.py --epochs 60 --batch_size 32

######################################

About

πŸ“Œ **15-Class Image Classifier** using EfficientNetB0 & TensorFlow/Keras A transfer learning model for multi-class image classification with data preprocessing, visualization & evaluation. πŸ”§ **Tech Stack**: - EfficientNetB0 (pretrained) - TensorFlow / Keras #ComputerVision #DeepLearning #TransferLearning

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published