Update 1.py

1.py

@@ -114,8 +114,8 @@ def merge_image_with_mask(image_input_path, detections, output_folder):
     dilated_mask = cv2.dilate(combined_mask, kernel, iterations=1)
 
     # 保存膨胀后的掩码
-    #mask_path = os.path.join(mask_folder, 'test1.png')
-    #cv2.imwrite(mask_path, dilated_mask * 255)
+    mask_path = os.path.join(mask_folder, image_name)
+    cv2.imwrite(mask_path, dilated_mask * 255)
 
     # 读取原始图像
     original_image = cv2.imread(image_input_path)