This repository contains a sample work to classify garbage type based on resized images on this repository.

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:

datagen = ImageDataGenerator(
    validation_split=0.2
)

While the trash-classification-aug.keras model trained with more data augmentation works in the dataset:

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

Downloads last month
35
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train diozhug/tensorflow-trash-classification