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.1.22 HUB model pathlib fix (#8621)

Browse Source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-03-04 01:08:41 +01:00 committed by GitHub
parent 946e18f79c
commit e138d701a0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 6 additions and 6 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.1.21" __version__ = "8.1.22"
from ultralytics.data.explorer.explorer import Explorer from ultralytics.data.explorer.explorer import Explorer
from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld

2
ultralytics/hub/__init__.py
View File

@ -48,7 +48,7 @@ def login(api_key: str = None, save=True) -> bool:
return True return True
else: else:
# Failed to authenticate with HUB # Failed to authenticate with HUB
LOGGER.info(f"{PREFIX}Retrieve API key from {api_key_url}") LOGGER.info(f"{PREFIX}Get API key from {api_key_url} and then run 'yolo hub login API_KEY'")
return False return False

2
ultralytics/hub/auth.py
View File

@ -64,7 +64,7 @@ class Auth:
if verbose: if verbose:
LOGGER.info(f"{PREFIX}New authentication successful ✅") LOGGER.info(f"{PREFIX}New authentication successful ✅")
elif verbose: elif verbose:
LOGGER.info(f"{PREFIX}Retrieve API key from {API_KEY_URL}") LOGGER.info(f"{PREFIX}Get API key from {API_KEY_URL} and then run 'yolo hub login API_KEY'")
def request_api_key(self, max_attempts=3): def request_api_key(self, max_attempts=3):
""" """

6
ultralytics/models/yolo/model.py
View File

@ -13,9 +13,9 @@ class YOLO(Model):
def __init__(self, model="yolov8n.pt", task=None, verbose=False): def __init__(self, model="yolov8n.pt", task=None, verbose=False):
"""Initialize YOLO model, switching to YOLOWorld if model filename contains '-world'.""" """Initialize YOLO model, switching to YOLOWorld if model filename contains '-world'."""
model = Path(model) path = Path(model)
if "-world" in model.stem and model.suffix in {".pt", ".yaml", ".yml"}: # if YOLOWorld PyTorch model if "-world" in path.stem and path.suffix in {".pt", ".yaml", ".yml"}: # if YOLOWorld PyTorch model
new_instance = YOLOWorld(model) new_instance = YOLOWorld(path)
self.__class__ = type(new_instance) self.__class__ = type(new_instance)
self.__dict__ = new_instance.__dict__ self.__dict__ = new_instance.__dict__
else: else: