From a6e2130c8b085521384044b8b409d7c8a9e2bf4d Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Wed, 30 Jun 2021 10:46:18 +0200
Subject: [PATCH] Added flexibilty regarding ssmi-import statement from skimage
in metrics.py.
---
.../model_modules/video_prediction/metrics.py | 11 ++++++++++-
1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/video_prediction_tools/model_modules/video_prediction/metrics.py b/video_prediction_tools/model_modules/video_prediction/metrics.py
index 253dfbbd..5b31919a 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])
--
GitLab