diff --git a/requirements.txt b/requirements.txt index c17b42c2..d219640d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ torchvision>=0.2.1 numpy>=1.14.3 scipy>=1.0.1 scikit-image>=0.13.0 -opencv>=2.4.11 +opencv-python>=2.4.11 matplotlib>=1.5.1 tqdm>=4.28.1 jupyter diff --git a/setup.py b/setup.py index 5a304dd8..b71a6d72 100644 --- a/setup.py +++ b/setup.py @@ -14,6 +14,7 @@ packages=['lpips'], package_data={'lpips': ['weights/v0.0/*.pth','weights/v0.1/*.pth']}, include_package_data=True, + install_requires=["torch>=0.4.0", "torchvision>=0.2.1", "numpy>=1.14.3", "scipy>=1.0.1", "tqdm>=4.28.1"], classifiers=[ "Programming Language :: Python :: 3", "License :: OSI Approved :: BSD License",