| | --- |
| | 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 |
