Skip to content

Commit

Permalink
Scope all hubconf.py imports for torch.hub.list() (ultralytics#3145)
Browse files Browse the repository at this point in the history
* Scope all hubconf.py imports for torch.hub.list()

* Update hubconf.py
  • Loading branch information
glenn-jocher committed May 12, 2021
1 parent 589fbe8 commit fe91134
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions hubconf.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,8 @@
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
"""

from pathlib import Path

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):
"""Creates a specified YOLOv5 model
Expand All @@ -29,11 +22,16 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
Returns:
YOLOv5 pytorch model
"""
from pathlib import Path

from models.yolo import Model, attempt_load
from utils.general import check_requirements, set_logging
from utils.google_utils import attempt_download
from utils.torch_utils import select_device

check_requirements(Path(__file__).parent / 'requirements.txt', exclude=('tensorboard', 'pycocotools', 'thop'))
set_logging(verbose=verbose)

fname = Path(name).with_suffix('.pt') # checkpoint filename
try:
if pretrained and channels == 3 and classes == 80:
Expand Down

0 comments on commit fe91134

Please sign in to comment.