huzey commited on
Commit
30e4b09
1 Parent(s): 0eae266

fix seg app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3365,9 +3365,9 @@ with demo:
3365
  # combine the masks, also draw the bounding boxes
3366
  combined_masks = []
3367
  for i_image in range(len(mask)):
3368
- noisy_mask = np.array(mask[i_image].convert("RGB"))
3369
  bbox = bboxs[i_image]
3370
- clean_mask = np.array(filtered_masks[i_image].convert("RGB"))
3371
  combined_mask = noisy_mask * 0.4 + clean_mask
3372
  combined_mask = np.clip(combined_mask, 0, 255).astype(np.uint8)
3373
  if overlay_image:
 
3365
  # combine the masks, also draw the bounding boxes
3366
  combined_masks = []
3367
  for i_image in range(len(mask)):
3368
+ noisy_mask = np.array(mask[i_image])
3369
  bbox = bboxs[i_image]
3370
+ clean_mask = np.array(filtered_masks[i_image])
3371
  combined_mask = noisy_mask * 0.4 + clean_mask
3372
  combined_mask = np.clip(combined_mask, 0, 255).astype(np.uint8)
3373
  if overlay_image: