Update app.py
Browse files
app.py
CHANGED
|
@@ -17,7 +17,11 @@ hidden_layer = model.layers[-2].output
|
|
| 17 |
model_new = Model(new_input, hidden_layer)
|
| 18 |
model = Sequential()
|
| 19 |
num_classes = 2
|
| 20 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
model.add(data_aug_layer)
|
| 22 |
model.add(model_imagenet)
|
| 23 |
|
|
|
|
| 17 |
model_new = Model(new_input, hidden_layer)
|
| 18 |
model = Sequential()
|
| 19 |
num_classes = 2
|
| 20 |
+
data_aug_layer = tf.keras.Sequential([
|
| 21 |
+
RandomFlip("horizontal"), # Rotate by up to 20 degrees
|
| 22 |
+
RandomZoom(0.2),
|
| 23 |
+
RandomRotation(0.1)
|
| 24 |
+
])
|
| 25 |
model.add(data_aug_layer)
|
| 26 |
model.add(model_imagenet)
|
| 27 |
|