Update main.py

main.py

@@ -12,6 +12,9 @@ from utils_mask import check_mask_overlap_torch, check_cover_all_torch, visualiz
 def run_main(
     name="example_tmp",
     name_2=None,
+    mask_list=None, 
+    mask_label_list=None,
+    image_gt=None,
     dpm="sd",
     resolution=512,
     seed=42,
@@ -72,12 +75,12 @@ def run_main(
 
     input_folder = os.path.join(base_input_folder, name)
 
-    mask_list, mask_label_list = load_mask(input_folder)
+    #mask_list, mask_label_list = load_mask(input_folder)
     assert mask_list[0].shape[0] == resolution, "Segmentation should be done on size {}".format(resolution)
-    try:
-        image_gt = load_image(os.path.join(input_folder, "img_{}.png".format(resolution) ), size = resolution)
-    except:
-        image_gt = load_image(os.path.join(input_folder, "img_{}.jpg".format(resolution) ), size = resolution)
+    #try:
+    #    image_gt = load_image(os.path.join(input_folder, "img_{}.png".format(resolution) ), size = resolution)
+    #except:
+    #    image_gt = load_image(os.path.join(input_folder, "img_{}.jpg".format(resolution) ), size = resolution)
 
     if image:
         input_folder_2 = os.path.join(base_input_folder, name_2)