README.md

---
datasets:
- garythung/trashnet
language:
- en
metrics:
- accuracy
library_name: keras
---
This repository contains a sample work to classify garbage type based on resized images on this [repository](https://huggingface.co/datasets/garythung/trashnet).

There are 2 models available:
- trash-classification-no-aug.keras
- trash-classification-aug.keras

The `trash-classification-no-aug.keras` model trained without data augmentation:
```python
datagen = ImageDataGenerator(
    validation_split=0.2
)
```
While the `trash-classification-aug.keras` model trained with more data augmentation works in the dataset:
```python
# With data augmentation
datagen = ImageDataGenerator(
    rescale=1./255, 
    validation_split=0.2,
    width_shift_range=0.1, 
    height_shift_range=0.1, 
    horizontal_flip=True
)
```

The models trained with Tensorflow Functional API by using this approach:
```
Conv --> BatchNorm --> Conv --> BatchNorm --> MaxPooling (3x)
```

For the detailed description about the training process and models' performace, you can visit this Github [repository](https://github.com/dioz95/trash-classification/tree/main).