🗑️ Garbage Classification Model — CS549

• Cardboard
• Glass
• Metal
• Paper
• Plastic
• Trash
• Biodegradable

🔍 Use Case

🧠 Model Overview

• Architecture: CNN (Conv2D, MaxPooling, Dense)
• Input: 224x224 RGB
• Output: 7-class probability vector
• Accuracy: ~92% (validation)
• Explainability: Supports Grad-CAM

📦 How to Use

from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np
[object Object]
[object Object]

pred = model.predict(img_array)
classes = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash', 'biodegradable']
print("Prediction:", classes[np.argmax(pred)])

📁 Files

• GarbageMLModel_CS549.h5 – Trained model
• label_map.json – Label mapping

👤 Author