xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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
Show Stats 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)