mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

PyTorch Hub and autoShape update (#1415)

pull/1420/head
Glenn Jocher 2020-11-16 23:14:07 +01:00
parent f5429260ca
commit 93fc607f07
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hubconf.py
View File

@ -14,7 +14,7 @@ from models.yolo import Model
from utils.general import set_logging from utils.general import set_logging
from utils.google_utils import attempt_download from utils.google_utils import attempt_download
dependencies = ['torch', 'yaml', 'pillow'] dependencies = ['torch', 'PyYAML>=5.3', 'Pillow']
set_logging() set_logging()

2
requirements.txt
View File

@ -5,7 +5,7 @@ Cython
matplotlib>=3.2.2 matplotlib>=3.2.2
numpy>=1.18.5 numpy>=1.18.5
opencv-python>=4.1.2 opencv-python>=4.1.2
pillow Pillow
PyYAML>=5.3 PyYAML>=5.3
scipy>=1.4.1 scipy>=1.4.1
tensorboard>=2.2 tensorboard>=2.2