diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index d1648f70..30e238b7 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.143' +__version__ = '8.0.144' from ultralytics.hub import start from ultralytics.models import RTDETR, SAM, YOLO diff --git a/ultralytics/models/sam/model.py b/ultralytics/models/sam/model.py index b941f7bf..44b32689 100644 --- a/ultralytics/models/sam/model.py +++ b/ultralytics/models/sam/model.py @@ -29,7 +29,7 @@ class SAM(Model): overrides = dict(conf=0.25, task='segment', mode='predict', imgsz=1024) kwargs.update(overrides) prompts = dict(bboxes=bboxes, points=points, labels=labels) - super().predict(source, stream, prompts=prompts, **kwargs) + return super().predict(source, stream, prompts=prompts, **kwargs) def __call__(self, source=None, stream=False, bboxes=None, points=None, labels=None, **kwargs): """Calls the 'predict' function with given arguments to perform object detection.""" diff --git a/ultralytics/models/sam/predict.py b/ultralytics/models/sam/predict.py index ebd072a7..396b3be3 100644 --- a/ultralytics/models/sam/predict.py +++ b/ultralytics/models/sam/predict.py @@ -294,7 +294,7 @@ class Predictor(BasePredictor): def setup_model(self, model, verbose=True): """Set up YOLO model with specified thresholds and device.""" - device = select_device(self.args.device) + device = select_device(self.args.device, verbose=verbose) if model is None: model = build_sam(self.args.model) model.eval()