Samarth991 commited on
Commit
9b2e4f0
·
1 Parent(s): e9d38e7

added segmentation mask overlay

Browse files
Files changed (1) hide show
  1. tool_utils/clip_segmentation.py +1 -0
tool_utils/clip_segmentation.py CHANGED
@@ -6,6 +6,7 @@ from transformers import CLIPSegProcessor, CLIPSegForImageSegmentation
6
  from segmentation_mask_overlay import overlay_masks
7
  from typing import List
8
  import logging
 
9
  class CLIPSEG:
10
  def __init__(self,model_name = "CIDAS/clipseg-rd64-refined",threshould=0.60):
11
  self.clip_processor = CLIPSegProcessor.from_pretrained(model_name)
 
6
  from segmentation_mask_overlay import overlay_masks
7
  from typing import List
8
  import logging
9
+
10
  class CLIPSEG:
11
  def __init__(self,model_name = "CIDAS/clipseg-rd64-refined",threshould=0.60):
12
  self.clip_processor = CLIPSegProcessor.from_pretrained(model_name)