|
|
---
|
|
|
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
|
|
|
|
