Update 2.py

2.py

@@ -116,8 +116,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)
@@ -131,7 +131,8 @@ def merge_image_with_mask(image_input_path, detections, output_folder):
     # 使用掩膜从原始图片中提取部分区域
     masked_image = cv2.bitwise_and(original_image, original_image, mask=dilated_mask)
     # 将掩膜应用于原始图片
-    blurred_image = cv2.GaussianBlur(original_image, (21, 21), 500)  # 使用较大的核大小进行模糊
+    #blurred_image = cv2.GaussianBlur(original_image, (21, 21), 500)  # 使用较大的核大小进行模糊
+    blurred_image = cv2.medianBlur(original_image, 21)
     # 将提取的部分区域叠加到模糊后的图片上
     blurred_image = cv2.bitwise_and(blurred_image, blurred_image, mask=~dilated_mask)
         # 将提取的部分区域叠加到模糊后的图片上