update
Browse files
app.py
CHANGED
|
@@ -13,6 +13,12 @@ resize_and_rescale = tf.keras.Sequential([
|
|
| 13 |
layers.Resizing(image_size, image_size),
|
| 14 |
layers.Rescaling(1.0/255)
|
| 15 |
])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
model = models.Sequential([
|
| 17 |
resize_and_rescale,
|
| 18 |
data_augmentation,
|
|
|
|
| 13 |
layers.Resizing(image_size, image_size),
|
| 14 |
layers.Rescaling(1.0/255)
|
| 15 |
])
|
| 16 |
+
|
| 17 |
+
#DATA AUGMENTATION
|
| 18 |
+
data_augmentation = tf.keras.Sequential([
|
| 19 |
+
layers.RandomFlip("horizontal_and_vertical"),
|
| 20 |
+
layers.RandomRotation(0.2)
|
| 21 |
+
])
|
| 22 |
model = models.Sequential([
|
| 23 |
resize_and_rescale,
|
| 24 |
data_augmentation,
|