metadata
tags:
- image-classification
- cnn
- waste-classification
license: mit
language:
- te
- en
WasteClassifier01 🗑️
A CNN model that classifies waste images into categories like plastic, paper, metal, glass, organic, etc.
Model Info
- Format: Keras
.h5 - Size: 17.3 MB
- Task: Image Classification
How to Load & Use
```python from tensorflow import keras import numpy as np from PIL import Image
Load model
model = keras.models.load_model("waste_classifier_final.h5")
Preprocess image
img = Image.open("your_image.jpg").resize((224, 224)) img_array = np.expand_dims(np.array(img) / 255.0, axis=0)
Predict
prediction = model.predict(img_array) print("Predicted class:", np.argmax(prediction)) ```
Author
dk00069