remove channel_axis=2 (NVIDIAGameWorks#139)

Signed-off-by: Clement Fuji Tsang <[email protected]>
AvivSham · May 30, 2023 · a5c6d5a · a5c6d5a
1 parent eb07b62
commit a5c6d5a
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,