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

use yolov10 by default in cli

Browse Source
This commit is contained in:
wa22 2024-06-04 11:06:48 +08:00
parent dfed201384
commit e5f2f03c55
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ultralytics/cfg/__init__.py
View File

@ -8,6 +8,7 @@ import sys
from pathlib import Path
from types import SimpleNamespace
from typing import Dict, List, Union
import re
from ultralytics.utils import (
ASSETS,
@ -551,7 +552,11 @@ def entrypoint(debug=""):
from ultralytics import SAM
model = SAM(model)
elif "yolov10" in stem:
elif re.search("v3|v5|v6|v8|v9", stem):
from ultralytics import YOLO
model = YOLO(model, task=task)
else:
from ultralytics import YOLOv10
# Special case for the HuggingFace Hub
@ -560,10 +565,6 @@ def entrypoint(debug=""):
model = YOLOv10.from_pretrained(model)
else:
model = YOLOv10(model)
else:
from ultralytics import YOLO
model = YOLO(model, task=task)
if isinstance(overrides.get("pretrained"), str):
model.load(overrides["pretrained"])