`ultralytics 8.0.139` add FastSAM CLI support (#3860)

single_channel
Glenn Jocher 1 year ago committed by GitHub
parent aa1cab74f8
commit ed25db9426
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = '8.0.138' __version__ = '8.0.139'
from ultralytics.engine.model import YOLO from ultralytics.engine.model import YOLO
from ultralytics.hub import start from ultralytics.hub import start

@ -372,6 +372,9 @@ def entrypoint(debug=''):
if 'rtdetr' in model.lower(): # guess architecture if 'rtdetr' in model.lower(): # guess architecture
from ultralytics import RTDETR from ultralytics import RTDETR
model = RTDETR(model) # no task argument model = RTDETR(model) # no task argument
elif 'fastsam' in model.lower():
from ultralytics import FastSAM
model = FastSAM(model)
elif 'sam' in model.lower(): elif 'sam' in model.lower():
from ultralytics import SAM from ultralytics import SAM
model = SAM(model) model = SAM(model)

Loading…
Cancel
Save