Fix SAM CLI stem usage (#6257)
This commit is contained in:
parent
ec977f6ca7
commit
87233ea17c
1 changed files with 4 additions and 3 deletions
|
|
@ -405,13 +405,14 @@ def entrypoint(debug=''):
|
|||
model = 'yolov8n.pt'
|
||||
LOGGER.warning(f"WARNING ⚠️ 'model' is missing. Using default 'model={model}'.")
|
||||
overrides['model'] = model
|
||||
if 'rtdetr' in model.lower(): # guess architecture
|
||||
stem = Path(model).stem.lower()
|
||||
if 'rtdetr' in stem: # guess architecture
|
||||
from ultralytics import RTDETR
|
||||
model = RTDETR(model) # no task argument
|
||||
elif 'fastsam' in model.lower():
|
||||
elif 'fastsam' in stem:
|
||||
from ultralytics import FastSAM
|
||||
model = FastSAM(model)
|
||||
elif 'sam' in model.lower():
|
||||
elif 'sam' in stem:
|
||||
from ultralytics import SAM
|
||||
model = SAM(model)
|
||||
else:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue