File size: 880 Bytes
ecde677 | 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 42 43 44 45 46 47 48 49 | ---
language: en
license: apache-2.0
library_name: tensorflow
pipeline_tag: image-classification
tags:
- waste-classification
- efficientnet
- recycling
- sustainability
---
# Waste Classifier (EfficientNetB0)
Image classification model for waste segregation into three classes:
- dry_waste
- wet_waste
- other_waste
## Model Files
- waste_classifier.keras: TensorFlow Keras model artifact
- inference.py: Local inference utility
- requirements.txt: Python dependencies for local inference
## Quick Start
```python
from inference import predict
result = predict("sample.jpg")
print(result)
```
## Output Format
```json
{
"class": "dry_waste",
"display_name": "♻️ Dry Waste",
"confidence": 92.5,
"action": "Can be recycled → Paper, plastic, metal recovery",
"all_predictions": {
"dry_waste": 92.5,
"other_waste": 4.1,
"wet_waste": 3.4
}
}
```
|