init
Browse files
app.py
CHANGED
|
@@ -6,7 +6,11 @@ image_size=256
|
|
| 6 |
channels=3
|
| 7 |
input_shape = (None, image_size, image_size, channels)
|
| 8 |
n_classes = 3
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
model = models.Sequential([
|
| 11 |
resize_and_rescale,
|
| 12 |
data_augmentation,
|
|
|
|
| 6 |
channels=3
|
| 7 |
input_shape = (None, image_size, image_size, channels)
|
| 8 |
n_classes = 3
|
| 9 |
+
#preprocessing during model creation RESCALING & RESIZING
|
| 10 |
+
resize_and_rescale = tf.keras.Sequential([
|
| 11 |
+
layers.Resizing(IMAGE_SIZE, IMAGE_SIZE),
|
| 12 |
+
layers.Rescaling(1.0/255)
|
| 13 |
+
])
|
| 14 |
model = models.Sequential([
|
| 15 |
resize_and_rescale,
|
| 16 |
data_augmentation,
|