Update app.py
Browse files
app.py
CHANGED
@@ -16,23 +16,23 @@ def classify_image(inp):
|
|
16 |
#import app
|
17 |
|
18 |
|
19 |
-
batch = 4
|
20 |
-
|
21 |
-
generator = ImageDataGenerator(
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
train_iterator = generator.flow_from_directory("C:/Users/Lyall Stewart/Documents/Coding/NeuralNetwork/COVID-19 Classification/data/train",
|
32 |
batch_size=batch,
|
33 |
color_mode='grayscale',
|
34 |
class_mode='sparse')
|
35 |
-
test_iterator = generator.flow_from_directory("C:/Users/Lyall Stewart/Documents/Coding/NeuralNetwork/COVID-19 Classification/data/test",
|
36 |
batch_size=batch,
|
37 |
color_mode='grayscale',
|
38 |
class_mode='sparse')
|
|
|
16 |
#import app
|
17 |
|
18 |
|
19 |
+
batch = 4
|
20 |
+
|
21 |
+
generator = ImageDataGenerator(
|
22 |
+
rotation_range=40,
|
23 |
+
width_shift_range=0.2,
|
24 |
+
height_shift_range=0.2,
|
25 |
+
rescale=1./255,
|
26 |
+
shear_range=0.2,
|
27 |
+
zoom_range=0.2,
|
28 |
+
horizontal_flip=True,
|
29 |
+
fill_mode='nearest')
|
30 |
+
|
31 |
+
train_iterator = generator.flow_from_directory("C:/Users/Lyall Stewart/Documents/Coding/NeuralNetwork/COVID-19 Classification/data/train",
|
32 |
batch_size=batch,
|
33 |
color_mode='grayscale',
|
34 |
class_mode='sparse')
|
35 |
+
test_iterator = generator.flow_from_directory("C:/Users/Lyall Stewart/Documents/Coding/NeuralNetwork/COVID-19 Classification/data/test",
|
36 |
batch_size=batch,
|
37 |
color_mode='grayscale',
|
38 |
class_mode='sparse')
|