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
