Unverified 提交 06372b14 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Scope all hubconf.py imports for torch.hub.list() (#3145)

* Scope all hubconf.py imports for torch.hub.list() * Update hubconf.py
上级 19352669
...@@ -5,15 +5,8 @@ Usage: ...@@ -5,15 +5,8 @@ Usage:
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
""" """
from pathlib import Path
import torch import torch
from utils.general import check_requirements, set_logging
dependencies = ['torch', 'yaml']
check_requirements(Path(__file__).parent / 'requirements.txt', exclude=('tensorboard', 'pycocotools', 'thop'))
def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbose=True): def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbose=True):
"""Creates a specified YOLOv5 model """Creates a specified YOLOv5 model
...@@ -29,11 +22,16 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo ...@@ -29,11 +22,16 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
Returns: Returns:
YOLOv5 pytorch model YOLOv5 pytorch model
""" """
from pathlib import Path
from models.yolo import Model, attempt_load from models.yolo import Model, attempt_load
from utils.general import check_requirements, set_logging
from utils.google_utils import attempt_download from utils.google_utils import attempt_download
from utils.torch_utils import select_device from utils.torch_utils import select_device
check_requirements(Path(__file__).parent / 'requirements.txt', exclude=('tensorboard', 'pycocotools', 'thop'))
set_logging(verbose=verbose) set_logging(verbose=verbose)
fname = Path(name).with_suffix('.pt') # checkpoint filename fname = Path(name).with_suffix('.pt') # checkpoint filename
try: try:
if pretrained and channels == 3 and classes == 80: if pretrained and channels == 3 and classes == 80:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论