PatchPrior

class deepinv.optim.PatchPrior(negative_patch_log_likelihood, n_patches=-1, patch_size=6, pad=False, *args, **kwargs)[source]

Bases: Prior

Patch prior \(g(x) = \sum_i h(P_i x)\) for some prior \(h(x)\) on the space of patches.

Given a negative log likelihood (NLL) function on the patch space, this builds a prior by summing the NLLs of all (overlapping) patches in the image.

Parameters:
  • negative_patch_log_likelihood (callable) – NLL function on the patch space

  • n_patches (int) – number of randomly selected patches for prior evaluation. -1 for taking all patches

  • patch_size (int) – size of the patches

  • pad (bool) – whether to use mirror padding on the boundary to avoid undesired boundary effects

g(x, *args, **kwargs)[source]

Computes the prior \(g(x)\).

Parameters:

x (torch.Tensor) – Variable \(x\) at which the prior is computed.

Returns:

(torch.tensor) prior \(g(x)\).

Examples using PatchPrior:

Patch priors for limited-angle computed tomography

Patch priors for limited-angle computed tomography