Update app.py
Browse files
app.py
CHANGED
|
@@ -9,13 +9,7 @@ from keras.applications.inception_v3 import InceptionV3
|
|
| 9 |
from keras.models import Model
|
| 10 |
|
| 11 |
model_imagenet = InceptionV3(weights='imagenet',include_top=False,input_shape=(180, 180, 3))
|
| 12 |
-
|
| 13 |
-
new_input = model.input
|
| 14 |
-
|
| 15 |
-
hidden_layer = model.layers[-2].output
|
| 16 |
-
|
| 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
|
|
|
|
| 9 |
from keras.models import Model
|
| 10 |
|
| 11 |
model_imagenet = InceptionV3(weights='imagenet',include_top=False,input_shape=(180, 180, 3))
|
| 12 |
+
model_imagenet.trainable = False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
num_classes = 2
|
| 14 |
data_aug_layer = tf.keras.Sequential([
|
| 15 |
RandomFlip("horizontal"), # Rotate by up to 20 degrees
|