---
license: mit
tags:
- trash-classification
- image-classification
- cnn
- svm
- knn
---

# Trash Classifier Models

This repository contains trained models for trash/material classification.

## Models Included

- **cnn_model.keras**: CNN model for feature extraction
- **svm_model.pkl**: SVM classifier
- **knn_model.pkl**: KNN classifier
- **scaler.pkl**: Feature scaler
- **class_mapping.pkl**: Class label mapping
- **config.json**: Model configuration

## Usage
```python
from huggingface_hub import hf_hub_download
import tensorflow as tf
import pickle

# Download and load CNN model
cnn_path = hf_hub_download(repo_id="moyasser20/trash-classifier-models", filename="cnn_model.keras")
cnn_model = tf.keras.models.load_model(cnn_path)

# Download and load SVM model
svm_path = hf_hub_download(repo_id="moyasser20/trash-classifier-models", filename="svm_model.pkl")
with open(svm_path, 'rb') as f:
    svm_model = pickle.load(f)

# Download other files similarly
```

## Training Details
- Task: Material/Trash Classification
- Framework: TensorFlow + Scikit-learn