diff --git a/video_prediction_tools/model_modules/video_prediction/metrics.py b/video_prediction_tools/model_modules/video_prediction/metrics.py index ef79a72dadef3651ec83607be7cd7e586d8cedef..f88fa38b9771a29c1ed5ec88ed9db417f270e31e 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 1fe4586428d5a639e4624f02b1fca3117faa292c..69d326f62c67c0931c3be818caecf94f4bdf5449 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