diff --git a/hubconf.py b/hubconf.py index 61ff4cc4f239..2266cab4c0e2 100644 --- 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 index d04638f18e87..c5e6a6119321 100755 --- 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