mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
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.
This commit is contained in:
parent
4d4a2b0520
commit
3f03acb3db
@ -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.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'))
|
check_requirements(requirements=Path(__file__).parent / 'requirements.txt',
|
||||||
|
exclude=('tensorboard', 'pycocotools', 'thop', 'opencv-python'))
|
||||||
set_logging(verbose=verbose)
|
set_logging(verbose=verbose)
|
||||||
|
|
||||||
fname = Path(name).with_suffix('.pt') # checkpoint filename
|
fname = Path(name).with_suffix('.pt') # checkpoint filename
|
||||||
|
Loading…
x
Reference in New Issue
Block a user