Spaces:
Running
Running
Commit
·
9b2e4f0
1
Parent(s):
e9d38e7
added segmentation mask overlay
Browse files
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)
|