File size: 1,381 Bytes
611e73e e6c10b0 611e73e cc45a67 611e73e e6c10b0 611e73e e6c10b0 611e73e e6c10b0 dcd643c e6c10b0 611e73e e6c10b0 611e73e | 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 50 51 52 53 54 55 | ---
license: mit
datasets:
- dvk65/TrashTypes
language:
- en
base_model:
- microsoft/resnet-50
---
This model is trained on a curated dataset of most frequently seen trash items in our college.</br>
## Model Details
- **Backbone**: ResNet50 (ImageNet pre-trained, fine-tuned)
- **Classes**: 13 trash / recycling / compost categories
- **Input size**: 224×224 RGB
- **Loss**: sparse_categorical_crossentropy
- **Optimizer**: Adam
## Dataset
Processed training, validation, and test splits are included in the `*_processed` directories.
Original dataset: [`dvk65/TrashTypes`](https://huggingface.co/dvk65/TrashTypes)
## Usage
```python
from huggingface_hub import hf_hub_download
import tensorflow as tf # tensorflow version above 2.20.0
REPO_ID = "dvk65/trash-classifier-resnet50"
FILENAME = "trashclassify_13.keras"
model_path = hf_hub_download(
repo_id=REPO_ID,
filename=FILENAME,
)
model = tf.keras.models.load_model(model_path)
```
The current target values are:
1. apples
2. bananas
3. bottles
4. cans
5. cardboard
6. cups
7. eggshells
8. mixed leftover food (labeled as generalcompost)
9. wooden coffee stirrers (labeled as mixers)
10. oranges (labeled as peels)
11. platicbags
12. plastic wrappers (labeled as plastics)
13. tissue papers
To help with expanding the dataset, feel free to contribute to: https://huggingface.co/datasets/dvk65/TrashTypes </br> |