Commit
·
d5063eb
1
Parent(s):
fe46428
Update eda.py
Browse files
eda.py
CHANGED
@@ -27,15 +27,17 @@ def run():
|
|
27 |
|
28 |
|
29 |
st.markdown('---')
|
30 |
-
|
31 |
|
32 |
# Define batch size and image size
|
33 |
batch_size = 256
|
34 |
img_size = (64, 64)
|
35 |
# Define paths to the data folders
|
36 |
-
|
37 |
-
|
38 |
-
|
|
|
|
|
39 |
# Create data generators for training, validation, and testing
|
40 |
train_datagen = ImageDataGenerator(
|
41 |
rescale=1./255,
|
|
|
27 |
|
28 |
|
29 |
st.markdown('---')
|
30 |
+
|
31 |
|
32 |
# Define batch size and image size
|
33 |
batch_size = 256
|
34 |
img_size = (64, 64)
|
35 |
# Define paths to the data folders
|
36 |
+
script_dir = os.path.dirname(os.path.abspath(__file__))
|
37 |
+
|
38 |
+
train_path = os.path.join(script_dir, 'food', 'Train')
|
39 |
+
valid_path = os.path.join(script_dir, 'food', 'Valid')
|
40 |
+
test_path = os.path.join(script_dir, 'food', 'Test')
|
41 |
# Create data generators for training, validation, and testing
|
42 |
train_datagen = ImageDataGenerator(
|
43 |
rescale=1./255,
|