Downloads · 30 days
8
0% of all-time downloads
trash-classifier-resnet50
dvk65/trash-classifier-resnet50
trash-classifier-resnet50 is a machine learning model from dvk65. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for keras. The card lists the license as mit.
This model is trained on a curated dataset of most frequently seen trash items in our college.</br
All-time downloads
3K
Public
Repo size
1.4 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.arrow938 MB · 81%
At a glance
- Library
- keras
- License
- mit
- Access
- Public
- Created
- Nov 19, 2025
- Updated
- Dec 9, 2025
- SHA
- 7a6e53ca
Base models
Datasets
Model card
From the Hugging Face model README
This model is trained on a curated dataset of most frequently seen trash items in our college.</br>
Model Details
- Backbone: ResNet50 (ImageNet pre-trained, fine-tuned)
- Classes: 13 trash / recycling / compost categories
- Input size: 224×224 RGB
- Loss: sparse_categorical_crossentropy
- Optimizer: Adam
Dataset
Processed training, validation, and test splits are included in the *_processed directories.
Original dataset: dvk65/TrashTypes
Usage
from huggingface_hub import hf_hub_download
import tensorflow as tf # tensorflow version above 2.20.0
REPO_ID = "dvk65/trash-classifier-resnet50"
FILENAME = "trashclassify_13.keras"
model_path = hf_hub_download(
repo_id=REPO_ID,
filename=FILENAME,
)
model = tf.keras.models.load_model(model_path)
The current target values are:
- apples
- bananas
- bottles
- cans
- cardboard
- cups
- eggshells
- mixed leftover food (labeled as generalcompost)
- wooden coffee stirrers (labeled as mixers)
- oranges (labeled as peels)
- platicbags
- plastic wrappers (labeled as plastics)
- tissue papers
To help with expanding the dataset, feel free to contribute to: https://huggingface.co/datasets/dvk65/TrashTypes </br>
- Library
- keras
- License
- mit
- Languages
- en
- Created
- Nov 19, 2025
- Updated
- Dec 9, 2025