Spaces:
Sleeping
Sleeping
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)
|