PnP_ADMM.py 🚧

Source: scripts/example_scripts/PnP_ADMM.py

Warning

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

device

experiment

lidc_testing_dataset

recon

diff

est_noise_std

denoiser

reconstructor

admm_iterations

admm_step_size

cg_max_iter

cg_tol

ssim_torch(target: torch.Tensor, input: torch.Tensor) → float

Adapt SSIM function for torch tensors.

denoiser_fn(ct_image: torch.Tensor, sigma: float = None) → torch.Tensor

No docstring is available.

denoiser_fn_admm(ct_image: torch.Tensor) → torch.Tensor

No docstring is available.