ssim.py 🚧

Source: LION/metrics/ssim.py

Warning

This file has not yet received a complete narrative and docstring audit. Its public source-level API is listed automatically below.

class SSIM

SSIM: A wrapper for skimage.metrics.structural_similarity that operates on (potentially) batched pytorch tensors

SSIM.forward(self, x: torch.Tensor, target: torch.Tensor, reduce=str | None, batched=True, channel_axis: int | None = 1) → torch.Tensor

No docstring is available.