LiamKhoaLe commited on
Commit
6b36bc1
·
1 Parent(s): 26110ee

Increase threshold to 50% of the garbage msk to lies on water

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -469,7 +469,7 @@ def _pipeline(uid,img_path):
469
  box_mask = movable_mask[y1:y2, x1:x2] # ← switch to movable_mask
470
  if box_mask.size == 0:
471
  continue
472
- if np.count_nonzero(box_mask) / box_mask.size >= 0.2:
473
  centres.append([int((x1 + x2) / 2), int((y1 + y2) / 2)])
474
  # add chunk centres and deduplicate
475
  centres.extend(chunk_centres)
 
469
  box_mask = movable_mask[y1:y2, x1:x2] # ← switch to movable_mask
470
  if box_mask.size == 0:
471
  continue
472
+ if np.count_nonzero(box_mask) / box_mask.size >= 0.5:
473
  centres.append([int((x1 + x2) / 2), int((y1 + y2) / 2)])
474
  # add chunk centres and deduplicate
475
  centres.extend(chunk_centres)