From ed25db94268fe0e65030f0ddc55e6180e39ffe81 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 21 Jul 2023 01:46:01 +0200 Subject: [PATCH] `ultralytics 8.0.139` add FastSAM CLI support (#3860) --- ultralytics/__init__.py | 2 +- ultralytics/cfg/__init__.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index ee89764..b387a07 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.138' +__version__ = '8.0.139' from ultralytics.engine.model import YOLO from ultralytics.hub import start diff --git a/ultralytics/cfg/__init__.py b/ultralytics/cfg/__init__.py index 75703de..65aae08 100644 --- a/ultralytics/cfg/__init__.py +++ b/ultralytics/cfg/__init__.py @@ -372,6 +372,9 @@ def entrypoint(debug=''): if 'rtdetr' in model.lower(): # guess architecture from ultralytics import RTDETR model = RTDETR(model) # no task argument + elif 'fastsam' in model.lower(): + from ultralytics import FastSAM + model = FastSAM(model) elif 'sam' in model.lower(): from ultralytics import SAM model = SAM(model)