HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.1.2 scope HUB-SDK imports (#7596)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-01-15 23:48:20 +01:00 • committed by GitHub
parent b1282544d2
commit 5f5f5d08f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 22 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

7
ultralytics/engine/model.py
View file

@ -5,11 +5,10 @@ import sys
from pathlib import Path
from typing import Union
from hub_sdk.config import HUB_WEB_ROOT
from ultralytics.cfg import TASK2DATA, get_cfg, get_save_dir
from ultralytics.nn.tasks import attempt_load_one_weight, guess_model_task, nn, yaml_model_load
from ultralytics.utils import ASSETS, DEFAULT_CFG_DICT, LOGGER, RANK, SETTINGS, callbacks, checks, emojis, yaml_load
from ultralytics.hub.utils import HUB_WEB_ROOT
class Model(nn.Module):
@ -123,9 +122,9 @@ class Model(nn.Module):
(
model.startswith(f"{HUB_WEB_ROOT}/models/"), # i.e. https://hub.ultralytics.com/models/MODEL_ID
[len(x) for x in model.split("_")] == [42, 20], # APIKEY_MODELID
len(model) == 20 and not Path(model).exists() and all(x not in model for x in "./\\"),
len(model) == 20 and not Path(model).exists() and all(x not in model for x in "./\\"), # MODELID
)
) # MODELID
)
def _new(self, cfg: str, task=None, model=None, verbose=True):
"""