[NVIDIAGameWorks#131] Fix skimage SSIM win_size error (NVIDIAGameWork…

…s#133) Addressing github issue NVIDIAGameWorks#131 : NVIDIAGameWorks#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.
AvivSham · May 30, 2023 · fb151c5 · fb151c5
1 parent 52e1d36
commit fb151c5
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)