From 3f03acb3dba3b4b3a4674fa9bdd6e73fbcbfae6c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 7 Jun 2021 15:38:51 +0200 Subject: [PATCH] check_requirements() exclude `opencv-python` (#3507) Duplicate of #3495 merged to `develop`. This PR will be merged to `master`. Fixes https://github.com/ultralytics/yolov5/issues/3494. --- hubconf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index f74e70c85a65..a52aae9fd1b7 100644 --- a/hubconf.py +++ b/hubconf.py @@ -30,7 +30,8 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo 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')) + check_requirements(requirements=Path(__file__).parent / 'requirements.txt', + exclude=('tensorboard', 'pycocotools', 'thop', 'opencv-python')) set_logging(verbose=verbose) fname = Path(name).with_suffix('.pt') # checkpoint filename