psnr.py 🚧

Source: LION/metrics/psnr.py

Warning

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

class PSNR

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

PSNR.forward(self, x: torch.Tensor, target: torch.Tensor, reduce=str | None, batched=True) → torch.Tensor

No docstring is available.