PyTorch Hub and autoShape update (#1415)

2020-11-16 23:14:07 +01:00 · 2020-11-16 23:14:07 +01:00 · 93fc607f07
parent f5429260ca
commit 93fc607f07
2 changed files with 2 additions and 2 deletions
--- a/hubconf.py
+++ b/hubconf.py
@ -14,7 +14,7 @@ from models.yolo import Model
 from utils.general import set_logging
 from utils.google_utils import attempt_download

-dependencies = ['torch', 'yaml', 'pillow']
+dependencies = ['torch', 'PyYAML>=5.3', 'Pillow']
 set_logging()


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