File size: 748 Bytes
7caa645
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
---
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