File size: 1,116 Bytes
30bb3a6 5489543 30bb3a6 5489543 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
---
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
|