diff --git a/video_prediction_tools/model_modules/video_prediction/metrics.py b/video_prediction_tools/model_modules/video_prediction/metrics.py index 253dfbbd0ad5881026f2e847064593f5f6b62119..5b31919a89dfa881940dc3ab991047c146cbf8bb 100644 --- a/video_prediction_tools/model_modules/video_prediction/metrics.py +++ b/video_prediction_tools/model_modules/video_prediction/metrics.py @@ -2,7 +2,16 @@ import tensorflow as tf #import lpips_tf import math import numpy as np -from skimage.measure import compare_ssim as ssim_ski +try: + from skimage.measure import compare_ssim as ssim_ski +except: + try: + import skimage.metrics._structural_similarity as ssmi_ski + except ModuleNotFoundError as err: + print("Could not get ssmi-function from skimage. Please check installed skimage-package.") + raise err + + def mse(a, b): return tf.reduce_mean(tf.squared_difference(a, b), [-3, -2, -1])