Upload app.py

app.py

@@ -425,11 +425,6 @@ def combine_mask(image, visible_mask_list, occlusion_mask_list):
     for mask in occlusion_mask_list:
         combined_occ_mask = cv2.bitwise_or(combined_mask, mask)
     # 添加 visible mask 边缘作为 occlusion mask 的一部分
-    # 1. 通过膨胀得到边缘区域：先膨胀，再减去原 mask 得到边界
-    kernel_edge = cv2.getStructuringElement(cv2.MORPH_RECT, (5, 5))
-    border_mask = cv2.dilate(combined_vis_mask, kernel_edge, iterations=1) - combined_vis_mask
-    # 2. 将边缘 mask 合并到 occlusion mask 中
-    combined_occ_mask = cv2.bitwise_or(combined_occ_mask, border_mask)
 
     overlay = apply_combined_mask_overlay(image, combined_vis_mask, combined_occ_mask)
     # 5*5 kernel dilate for occlusion mask
@@ -438,7 +433,7 @@ def combine_mask(image, visible_mask_list, occlusion_mask_list):
     combined_mask[combined_occ_mask > 0] = 128
     combined_mask[combined_vis_mask > 0] = 255
     # concat the mask and overlay to be a single image
-    print(overlay.shape, combined_mask.shape)
+    # print(overlay.shape, combined_mask.shape)
     result = cv2.hconcat([overlay, combined_mask[..., None].repeat(3, axis=-1)])
     return result