Update 2.py

2.py

@@ -81,6 +81,16 @@ import cv2
 import os
 from PIL import Image, ImageFilter
 
+mask_folder = 'mask2'
+if not os.path.exists(mask_folder):
+    os.makedirs(mask_folder)
+shutil.rmtree('mask2')
+mask_folder = 'mask2'
+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,8 +102,7 @@ def merge_image_with_mask(image_input_path, detections, output_folder):
 
     # 创建掩码文件夹
     mask_folder = 'mask2'
-    if not os.path.exists(mask_folder):
-        os.makedirs(mask_folder)
+
 
     # 合并掩码
     combined_mask = np.zeros_like(detections.mask[0], dtype=np.uint8)