fix imports

saicinpainting/evaluation/losses/base_loss.py

@@ -2,8 +2,6 @@ import logging
 from abc import abstractmethod, ABC
 
 import numpy as np
-import sklearn
-import sklearn.svm
 import torch
 import torch.nn as nn
 import torch.nn.functional as F

saicinpainting/evaluation/losses/lpips.py

@@ -56,7 +56,7 @@ def psnr(p0, p1, peak=255.):
 
 
 def dssim(p0, p1, range=255.):
-    return (1 - compare_ssim(p0, p1, data_range=range, multichannel=True)) / 2.
+    return (1 - structural_similarity(p0, p1, data_range=range, multichannel=True)) / 2.
 
 
 def rgb2lab(in_img, mean_cent=False):