PyTorch Hub and autoShape update (#1415)

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()
 
 

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