You can deploy this model and start earning money today!

Stefaron/
trash-classifier

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

Name: trash-classifier
Rating: 2.5 (12 reviews)

Views

No views yet

Trash Classifier Model

This model is a fine-tuned ResNet50 for classifying trash images into categories such as plastic, paper, glass, metal, and others.

Load the model and use it for inference:

python

import tensorflow as tf
model = tf.keras.models.load_model("Stefaron/trash-classifier/best_model.keras")