🚗 Car Classification Model (196 Classes)

📌 Overview

🧠 Model Architecture

• Backbone: EfficientNetB3
• Input Shape: (224, 224, 3)
• Output: 196 classes (softmax)
• Total Parameters: 11,755,251

Key Components:

• Custom Layers:
  • CastToFloat32
  • EfficientNetPreprocess
  • GeMPooling
• Data Augmentation:
  • Random Flip
  • Rotation
  • Zoom
  • Brightness & Contrast adjustment
  • Translation
• Fully Connected Head:
  • Dense → BatchNorm → ReLU → Dropout
  • Final Softmax layer

📊 Performance

• Accuracy: 80%

📂 Model Details

• Weights file: final_cars.keras
• Model is reconstructed programmatically and weights are loaded separately to avoid serialization issues with Lambda layers.

⚙️ Usage

1. Load Model

2. Inference Example

🏗️ How It Works

1. Input image is augmented
2. Converted to float32
3. Preprocessed using EfficientNet preprocessing
4. Passed through EfficientNetB3 backbone
5. Features pooled using GeM pooling
6. Fully connected layers perform classification

💡 Highlights

• Avoids Lambda layer serialization issues using custom registered layers
• Uses GeM pooling instead of traditional average pooling
• Strong augmentation pipeline improves generalization

📌 Notes

• Ensure TensorFlow version compatibility when loading weights
• Model expects input images resized to 224×224

👤 Author