Commit
·
fe46428
1
Parent(s):
b66900b
Update prediction.py
Browse files- prediction.py +3 -3
prediction.py
CHANGED
@@ -26,9 +26,9 @@ main_path= '\\food'
|
|
26 |
batch_size = 256
|
27 |
img_size = (64, 64)
|
28 |
# Define paths to the data folders
|
29 |
-
train_path =
|
30 |
-
valid_path =
|
31 |
-
test_path =
|
32 |
|
33 |
# Create data generators for training, validation, and testing
|
34 |
train_datagen = ImageDataGenerator(
|
|
|
26 |
batch_size = 256
|
27 |
img_size = (64, 64)
|
28 |
# Define paths to the data folders
|
29 |
+
train_path = '\\food\\Train'
|
30 |
+
valid_path = '\\food\\Valid'
|
31 |
+
test_path = '\\food\\Test'
|
32 |
|
33 |
# Create data generators for training, validation, and testing
|
34 |
train_datagen = ImageDataGenerator(
|