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

Author: Richard Zhang

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

requirements.txt

@@ -6,3 +6,4 @@ scikit-image>=0.13.0
 opencv>=2.4.11
 matplotlib>=1.5.1
 tqdm>=4.28.1
+jupyter

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",