diff --git a/src/loss/perceptual_similarity/perceptual_loss.py b/src/loss/perceptual_similarity/perceptual_loss.py index 7d87d6d..1babd5c 100644 --- a/src/loss/perceptual_similarity/perceptual_loss.py +++ b/src/loss/perceptual_similarity/perceptual_loss.py @@ -4,7 +4,7 @@ from __future__ import print_function import numpy as np -from skimage.measure import compare_ssim +from skimage.metrics import structural_similarity as compare_ssim import torch from torch.autograd import Variable