diff --git a/Inference.py b/Inference.py index 61b70fc..12c9f50 100644 --- a/Inference.py +++ b/Inference.py @@ -4,6 +4,7 @@ import torch from PIL import Image from utils.tools import convert_box_xywh_to_xyxy +import clip def parse_args(): diff --git a/fastsam/prompt.py b/fastsam/prompt.py index 4a2b900..176c02e 100644 --- a/fastsam/prompt.py +++ b/fastsam/prompt.py @@ -6,6 +6,7 @@ import torch from .utils import image_to_np_ndarray from PIL import Image +import clip class FastSAMPrompt: