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

…arity
richzhang · Sep 28, 2020 · af38ce5 · af38ce5
2 parents 43da2a5 + 5ec1e75
commit af38ce5
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/lpips/trainer.py b/lpips/trainer.py
@@ -9,6 +9,8 @@
 from scipy.ndimage import zoom
 from tqdm import tqdm
 import lpips
+import os
+
 
 class Trainer():
  def name(self):
@@ -62,7 +64,7 @@ def initialize(self, model='lpips', net='alex', colorspace='Lab', pnet_rand=Fals
 
  if self.is_train: # training mode
  # extra network on top to go from distances (d0,d1) => predicted human judgment (h*)
- self.rankLoss = networks.BCERankingLoss()
+ self.rankLoss = lpips.BCERankingLoss()
  self.parameters += list(self.rankLoss.net.parameters())
  self.lr = lr
  self.old_lr = lr

diff --git a/requirements.txt b/requirements.txt
@@ -6,3 +6,4 @@ scikit-image>=0.13.0
 opencv>=2.4.11
 matplotlib>=1.5.1
 tqdm>=4.28.1
+jupyter
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
  long_description = fh.read()
 setuptools.setup(
  name='lpips', 
- version='0.1',
+ version='0.1.2',
  author="Richard Zhang",
  author_email="[email protected]",
  description="LPIPS Similarity metric",