From 406f14a34ab2f5b23b8994afec2eaa3f86c69d29 Mon Sep 17 00:00:00 2001
From: gong1 <b.gong@fz-juelich.de>
Date: Mon, 24 May 2021 19:03:13 +0200
Subject: [PATCH] address the issue for import skimage metrics
---
.../model_modules/video_prediction/metrics.py | 2 +-
video_prediction_tools/postprocess/statistical_evaluation.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/video_prediction_tools/model_modules/video_prediction/metrics.py b/video_prediction_tools/model_modules/video_prediction/metrics.py
index ef79a72d..f88fa38b 100644
--- a/video_prediction_tools/model_modules/video_prediction/metrics.py
+++ b/video_prediction_tools/model_modules/video_prediction/metrics.py
@@ -1,7 +1,7 @@
import tensorflow as tf
#import lpips_tf
import math
-from skimage.metrics import structural_similarity as ssim_ski
+from skimage.measure import compare_ssim as ssim_ski
def mse(a, b):
return tf.reduce_mean(tf.squared_difference(a, b), [-3, -2, -1])
diff --git a/video_prediction_tools/postprocess/statistical_evaluation.py b/video_prediction_tools/postprocess/statistical_evaluation.py
index 1fe45864..69d326f6 100644
--- a/video_prediction_tools/postprocess/statistical_evaluation.py
+++ b/video_prediction_tools/postprocess/statistical_evaluation.py
@@ -2,7 +2,7 @@
import numpy as np
import xarray as xr
from typing import Union, List
-from skimage.metrics import structural_similarity as ssim
+from skimage.measure import compare_ssim as ssim
try:
from tqdm import tqdm
--
GitLab