setup

setup.py

@@ -0,0 +1,22 @@
+
+import setuptools
+with open("README.md", "r") as fh:
+ long_description = fh.read()
+setuptools.setup(
+ name='lpips', 
+ version='0.1',
+ author="Richard Zhang",
+ author_email="[email protected]",
+ description="LPIPS Similarity metric",
+ long_description=long_description,
+ long_description_content_type="text/markdown",
+ url="https://github.com/richzhang/PerceptualSimilarity",
+ packages=['lpips'],
+ package_data={'lpips': ['weights/v0.0/*.pth','weights/v0.1/*.pth']},
+ include_package_data=True,
+ classifiers=[
+ "Programming Language :: Python :: 3",
+ "License :: OSI Approved :: BSD License",
+ "Operating System :: OS Independent",
+ ],
+ )

test_dataset_model.py

@@ -43,9 +43,9 @@
 
  # evaluate model on data
  if(opt.dataset_mode=='2afc'):
- (score, results_verbose) = dm.score_2afc_dataset(data_loader, trainer.forward, name=dataset)
+ (score, results_verbose) = lpips.score_2afc_dataset(data_loader, trainer.forward, name=dataset)
  elif(opt.dataset_mode=='jnd'):
- (score, results_verbose) = dm.score_jnd_dataset(data_loader, trainer.forward, name=dataset)
+ (score, results_verbose) = lpips.score_jnd_dataset(data_loader, trainer.forward, name=dataset)
 
  # print results
  print(' Dataset [%s]: %.2f'%(dataset,100.*score))