Create README.md

README.md

@@ -0,0 +1,31 @@
+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 with no (or minimum) data augmentation:
+```python
+datagen = ImageDataGenerator(
+    rescale=1./255, 
+    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).