ultralytics 8.2.84 new SAM flexible imgsz inference (#15882)

Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-08-30 20:59:08 +08:00 · 2024-08-30 20:59:08 +08:00 · 7053169fd0
commit 7053169fd0
parent 5d66140ce1
6 changed files with 70 additions and 7 deletions
--- a/ultralytics/models/sam/model.py
+++ b/ultralytics/models/sam/model.py
@ -106,7 +106,7 @@ class SAM(Model):
            ...     print(f"Detected {len(r.masks)} masks")
        """
        overrides = dict(conf=0.25, task="segment", mode="predict", imgsz=1024)
-        kwargs.update(overrides)
+        kwargs = {**overrides, **kwargs}
        prompts = dict(bboxes=bboxes, points=points, labels=labels)
        return super().predict(source, stream, prompts=prompts, **kwargs)