metadata
license: mit
datasets:
- garythung/trashnet
language:
- en
metrics:
- accuracy
- confusion_matrix
- f1
Trash Classifier Model
This model is a fine-tuned ResNet50 for classifying trash images into categories such as plastic, paper, glass, metal, and others.
Dataset
- Dataset: TrashNet
- Number of Classes: 6 (Plastic, Metal, Paper, Glass, Cardboard, Trash)
Model Details
- Base Model: ResNet50 (pretrained on ImageNet)
- Fine-tuning: Input layer adjusted for 6 classes.
- Input Shape: (384, 384, 3)
- Learning Rate: 0.0001
- Optimizer: Adam
Performance
- Accuracy F1-Scores: 92%
How to Use
Load the model and use it for inference:
import tensorflow as tf
model = tf.keras.models.load_model("Stefaron/trash-classifier/best_model.keras")