ultralytics 8.0.139 add FastSAM CLI support (#3860)

2023-07-21 01:46:01 +02:00
parent aa1cab74f8
commit ed25db9426
2 changed files with 4 additions and 1 deletions
--- 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)