jackyccl commited on
Commit
99c8a45
1 Parent(s): dc4f951

Fixed remove bb box and extended mask

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -241,6 +241,15 @@ def lama_cleaner_process(image, mask):
241
  image = Image.open(io.BytesIO(numpy_to_bytes(res_np_img, 'png')))
242
  return image
243
 
 
 
 
 
 
 
 
 
 
244
  def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_threshold, text_threshold,
245
  iou_threshold, inpaint_mode, mask_source_radio, remove_mode, remove_mask_extend):
246
 
@@ -392,7 +401,8 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
392
  except:
393
  remove_mask_extend = 10
394
  mask_pil_exp = mask_extend(copy.deepcopy(mask_pil).convert("RGB"),
395
- box_convert(torch.tensor(boxes_filt_ori_array[i]), in_fmt="cxcywh", out_fmt="xyxy").numpy(),
 
396
  extend_pixels=remove_mask_extend, useRectangle=useRectangle)
397
  mask_imgs.append(mask_pil_exp)
398
  mask_pil = mix_masks(mask_imgs)
 
241
  image = Image.open(io.BytesIO(numpy_to_bytes(res_np_img, 'png')))
242
  return image
243
 
244
+ def xywh_to_xyxy(box, sizeW, sizeH):
245
+ if isinstance(box, list):
246
+ box = torch.Tensor(box)
247
+ box = box * torch.Tensor([sizeW, sizeH, sizeW, sizeH])
248
+ box[:2] -= box[2:] / 2 # top left corner
249
+ box[2:] += box[:2] # bottom right corner
250
+ box = box.numpy()
251
+ return box
252
+
253
  def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_threshold, text_threshold,
254
  iou_threshold, inpaint_mode, mask_source_radio, remove_mode, remove_mask_extend):
255
 
 
401
  except:
402
  remove_mask_extend = 10
403
  mask_pil_exp = mask_extend(copy.deepcopy(mask_pil).convert("RGB"),
404
+ # box_convert(torch.tensor(boxes_filt_ori_array[i]), in_fmt="cxcywh", out_fmt="xyxy").numpy(),
405
+ xywh_to_xyxy(torch.tensor(boxes_filt_ori_array[i]), size[0], size[1]),
406
  extend_pixels=remove_mask_extend, useRectangle=useRectangle)
407
  mask_imgs.append(mask_pil_exp)
408
  mask_pil = mix_masks(mask_imgs)