HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.0.139 add FastSAM CLI support (#3860)

Browse source
This commit is contained in:
Glenn Jocher 2023-07-21 01:46:01 +02:00 • committed by GitHub
parent aa1cab74f8
commit ed25db9426
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/__init__.py
View file

@ -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

3
ultralytics/cfg/__init__.py
View file

@ -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)