YOLOv5 PyTorch Hub models >> check_requirements() (#2588)
Browse files* YOLOv5 PyTorch Hub models >> check_requirements()
Update YOLOv5 PyTorch Hub requirements.txt path to cache path.
* Update hubconf.py
- hubconf.py +1 -1
hubconf.py
CHANGED
@@ -15,7 +15,7 @@ from utils.google_utils import attempt_download
|
|
15 |
from utils.torch_utils import select_device
|
16 |
|
17 |
dependencies = ['torch', 'yaml']
|
18 |
-
check_requirements(exclude=('pycocotools', 'thop'))
|
19 |
set_logging()
|
20 |
|
21 |
|
|
|
15 |
from utils.torch_utils import select_device
|
16 |
|
17 |
dependencies = ['torch', 'yaml']
|
18 |
+
check_requirements(Path(__file__).parent / 'requirements.txt', exclude=('pycocotools', 'thop'))
|
19 |
set_logging()
|
20 |
|
21 |
|