yizhangliu commited on
Commit
db51c0a
1 Parent(s): c2a6c29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -663,6 +663,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
663
  boxes_filt[i][:2] -= boxes_filt[i][2:] / 2
664
  boxes_filt[i][2:] += boxes_filt[i][:2]
665
 
 
666
  boxes_filt = boxes_filt.to(sam_device)
667
  transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])
668
 
@@ -702,6 +703,9 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
702
  mask_pil = input_mask_pil
703
  mask = input_mask
704
  else:
 
 
 
705
  masks_ori = copy.deepcopy(masks)
706
  if inpaint_mode == 'merge':
707
  masks = torch.sum(masks, dim=0).unsqueeze(0)
 
663
  boxes_filt[i][:2] -= boxes_filt[i][2:] / 2
664
  boxes_filt[i][2:] += boxes_filt[i][:2]
665
 
666
+ logger.info(f'boxes_filt={boxes_filt}')
667
  boxes_filt = boxes_filt.to(sam_device)
668
  transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])
669
 
 
703
  mask_pil = input_mask_pil
704
  mask = input_mask
705
  else:
706
+ logger.info(f'masks_shape={masks.shape}')
707
+ logger.info(f'masks={masks}')
708
+
709
  masks_ori = copy.deepcopy(masks)
710
  if inpaint_mode == 'merge':
711
  masks = torch.sum(masks, dim=0).unsqueeze(0)