dk00069 commited on
Commit
7caa645
·
verified ·
1 Parent(s): f7bbbdc

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - image-classification
4
+ - cnn
5
+ - waste-classification
6
+ license: mit
7
+ language:
8
+ - te
9
+ - en
10
+ ---
11
+
12
+ # WasteClassifier01 🗑️
13
+
14
+ A CNN model that classifies waste images into categories
15
+ like plastic, paper, metal, glass, organic, etc.
16
+
17
+ ## Model Info
18
+ - Format: Keras `.h5`
19
+ - Size: 17.3 MB
20
+ - Task: Image Classification
21
+
22
+ ## How to Load & Use
23
+ \```python
24
+ from tensorflow import keras
25
+ import numpy as np
26
+ from PIL import Image
27
+
28
+ # Load model
29
+ model = keras.models.load_model("waste_classifier_final.h5")
30
+
31
+ # Preprocess image
32
+ img = Image.open("your_image.jpg").resize((224, 224))
33
+ img_array = np.expand_dims(np.array(img) / 255.0, axis=0)
34
+
35
+ # Predict
36
+ prediction = model.predict(img_array)
37
+ print("Predicted class:", np.argmax(prediction))
38
+ \```
39
+
40
+ ## Author
41
+ dk00069