mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

YOLOv5 PyTorch Hub models >> check_requirements() (#2577)

Browse Source 
* Update hubconf.py with check_requirements()

Dependency checks have been missing from YOLOv5 PyTorch Hub model loading, causing errors in some cases when users are attempting to import hub models in unsupported environments. This should examine the YOLOv5 requirements.txt file and pip install any missing or version-conflict packages encountered. 

This is highly experimental (!), please let us know if this creates problems in your custom workflows.

* Update hubconf.py
This commit is contained in:
Glenn Jocher 2021-03-24 15:42:00 +01:00 committed by GitHub
parent e5b0200cd2
commit 2bcc89d762
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
hubconf.py
View File

@ -1,8 +1,8 @@
"""File for accessing YOLOv5 via PyTorch Hub https://pytorch.org/hub/ """File for accessing YOLOv5 models via PyTorch Hub https://pytorch.org/hub/ultralytics_yolov5/
Usage: Usage:
import torch import torch
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80) model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
""" """
from pathlib import Path from pathlib import Path
@ -10,11 +10,12 @@ from pathlib import Path
import torch import torch
from models.yolo import Model from models.yolo import Model
from utils.general import set_logging 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
dependencies = ['torch', 'yaml'] dependencies = ['torch', 'yaml']
check_requirements(exclude=('pycocotools', 'thop'))
set_logging() set_logging()