PyTorch Hub and autoShape update (ultralytics#1415)

fhkiel-mlaip · Nov 16, 2020 · cf6d67e · cf6d67e
1 parent 822b604
commit cf6d67e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/hubconf.py b/hubconf.py
@@ -14,7 +14,7 @@
 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()
 
 

diff --git 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