Upload app.py

app.py

@@ -327,9 +327,10 @@ def check_combined_mask(image, visibility_mask, visibility_mask_list, occlusion_
         boundary_mask = dilated - combined_mask
         occluded_mask = cv2.bitwise_or(occluded_mask, boundary_mask)
         occluded_mask = (occluded_mask > 0).astype(np.uint8)
-
-    occluded_mask = cv2.dilate(occluded_mask, kernel_small, iterations=1)
-    occluded_mask = (occluded_mask > 0).astype(np.uint8)
+        occluded_mask = cv2.dilate(occluded_mask, kernel_small, iterations=1)
+        occluded_mask = (occluded_mask > 0).astype(np.uint8)
+    else:
+        occluded_mask = 1 - combined_mask
 
     combined_mask[occluded_mask == 1] = 0