ultralytics 8.0.144 fix SAM predict() results (#4027)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-29 22:37:59 +08:00 · 2023-07-29 22:37:59 +08:00 · dbdea24955
commit dbdea24955
parent b3ddd9d09c
3 changed files with 3 additions and 3 deletions
--- 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()