Merge branch 'master' of https://github.com/richzhang/PerceptualSimil…

…arity
richzhang · Aug 25, 2021 · 3ba6e37 · 3ba6e37
2 parents 1c4cf7c + 3dd44f5
commit 3ba6e37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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",