remove channel_axis=2 (#139)

Signed-off-by: Clement Fuji Tsang <cfujitsang@nvidia.com>
NVIDIAGameWorks · Apr 5, 2023 · 6e2e72c · 6e2e72c
1 parent 78e1be1
commit 6e2e72c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/wisp/ops/image/metrics.py b/wisp/ops/image/metrics.py
@@ -84,7 +84,6 @@ def ssim(rgb, gts):
     return skimage.metrics.structural_similarity(
         rgb[..., :3].cpu().numpy(),
         gts[..., :3].cpu().numpy(),
-        channel_axis=2,
         data_range=1,
         gaussian_weights=True,
         sigma=1.5,