Update 1.py

1.py

@@ -81,6 +81,14 @@ import cv2
 import os
 from PIL import Image, ImageFilter
 
+mask_folder = 'mask1'
+if not os.path.exists(mask_folder):
+    os.makedirs(mask_folder)
+shutil.rmtree('mask1')
+mask_folder = 'mask1'
+if not os.path.exists(mask_folder):
+    os.makedirs(mask_folder)
+
 def merge_image_with_mask(image_input_path, detections, output_folder):
     # 创建输出文件夹
     if not os.path.exists(output_folder):
@@ -92,12 +100,7 @@ def merge_image_with_mask(image_input_path, detections, output_folder):
 
     # 创建掩码文件夹
     mask_folder = 'mask1'
-    if not os.path.exists(mask_folder):
-        os.makedirs(mask_folder)
-    shutil.rmtree('mask1')
-    mask_folder = 'mask1'
-    if not os.path.exists(mask_folder):
-        os.makedirs(mask_folder)
+
 
     # 合并掩码
     combined_mask = np.zeros_like(detections.mask[0], dtype=np.uint8)