From 28a99641c07c311aec98d54e65ed9a807ce4a59f Mon Sep 17 00:00:00 2001
From: Jaeheung Surh <jaeheungs@users.noreply.github.com>
Date: Fri, 24 Mar 2023 03:56:31 +0900
Subject: [PATCH] [#131] Fix skimage SSIM win_size error (#133)

Addressing github issue #131 : https://github.com/NVIDIAGameWorks/kaolin-wisp/issues/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.
---
 wisp/ops/image/metrics.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/wisp/ops/image/metrics.py b/wisp/ops/image/metrics.py
index 0f971c7..1454ea0 100644
--- 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)