Pavan2k4 commited on
Commit
9a4f3b6
·
verified ·
1 Parent(s): 5f4ed24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -19,15 +19,16 @@ from Utils.split_merge import split, merge
19
  from Utils.convert import read_pansharpened_rgb
20
  import shutil
21
 
 
22
  # Define directories
23
  UPLOAD_DIR = "data/uploaded_images/"
24
  MASK_DIR = "data/generated_masks/"
25
- PATCHES_DIR = 'data/Patches/'
26
- PRED_PATCHES_DIR = 'data/Patch_pred/'
27
  CSV_LOG_PATH = "image_log.csv"
28
 
29
  # Create directories
30
- for directory in [UPLOAD_DIR, MASK_DIR, PATCHES_DIR, PRED_PATCHES_DIR]:
31
  os.makedirs(directory, exist_ok=True)
32
 
33
  # Load model
 
19
  from Utils.convert import read_pansharpened_rgb
20
  import shutil
21
 
22
+
23
  # Define directories
24
  UPLOAD_DIR = "data/uploaded_images/"
25
  MASK_DIR = "data/generated_masks/"
26
+ patches_folder = 'data/Patches/'
27
+ pred_patches = 'data/Patch_pred/'
28
  CSV_LOG_PATH = "image_log.csv"
29
 
30
  # Create directories
31
+ for directory in [UPLOAD_DIR, MASK_DIR, patches_folder, pred_patches]:
32
  os.makedirs(directory, exist_ok=True)
33
 
34
  # Load model