PnP#

class deepinv.optim.PnP(denoiser, *args, **kwargs)[source]#

Plug-and-play prior \(\operatorname{prox}_{\gamma \regname}(x) = \operatorname{D}_{\sigma}(x)\).

Parameters:: denoiser (Callable) – Denoiser \(\operatorname{D}_{\sigma}\).

prox(x, sigma_denoiser, *args, **kwargs)[source]#

Uses denoising as the proximity operator of the PnP prior \(\regname\) at \(x\).

Parameters:

x (torch.Tensor) – Variable \(x\) at which the proximity operator is computed.
sigma_denoiser (float) – noise level parameter of the denoiser.

Returns:

(torch.tensor) proximity operator at \(x\).

Examples using `PnP`:#