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