xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +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 pathlib import Path
from types import SimpleNamespace from types import SimpleNamespace
from typing import Dict, List, Union from typing import Dict, List, Union
import re
from ultralytics.utils import ( from ultralytics.utils import (
ASSETS, ASSETS,
@ -551,7 +552,11 @@ def entrypoint(debug=""):
from ultralytics import SAM from ultralytics import SAM
model = SAM(model) 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 from ultralytics import YOLOv10
# Special case for the HuggingFace Hub # Special case for the HuggingFace Hub
@ -560,10 +565,6 @@ def entrypoint(debug=""):
model = YOLOv10.from_pretrained(model) model = YOLOv10.from_pretrained(model)
else: else:
model = YOLOv10(model) model = YOLOv10(model)
else:
from ultralytics import YOLO
model = YOLO(model, task=task)
if isinstance(overrides.get("pretrained"), str): if isinstance(overrides.get("pretrained"), str):
model.load(overrides["pretrained"]) model.load(overrides["pretrained"])