[#131] Fix skimage SSIM win_size error (#133)

Addressing github issue #131 : #131 Fixes: the demos on the README leads to an error due to skimage thinking that the color channel for the images used for SSIM are part of the image dimension rather than channels.
NVIDIAGameWorks · Mar 23, 2023 · 28a9964 · 28a9964
1 parent fc2d451
commit 28a9964
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wisp/ops/image/metrics.py b/wisp/ops/image/metrics.py
@@ -87,4 +87,5 @@ def ssim(rgb, gts):
         multichannel=True,
         data_range=1,
         gaussian_weights=True,
-        sigma=1.5)
+        sigma=1.5,
+        channel_axis=-1)