diozhug
/

tensorflow-trash-classification

Model card Files Files and versions

diozhug commited on May 22, 2024

Commit

629793e

·

verified ·

1 Parent(s): 02033fb

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -13,10 +13,9 @@ 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
 )
 ```

 - 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
 )
 ```