plot_curves#

deepinv.utils.plot_curves(metrics, save_dir=None, show=True)[source]#

Plots the metrics of a Plug-and-Play algorithm.

Parameters:

  • metrics (dict) – dictionary of metrics to plot.

  • save_dir (str) – path to save the plot.

  • show (bool) – show the image plot.

Examples using plot_curves:#

Distributed Plug-and-Play (PnP) Reconstruction

Distributed Plug-and-Play (PnP) Reconstruction

Radio interferometric imaging with deepinverse

Radio interferometric imaging with deepinverse

Image deblurring with Total-Variation (TV) prior

Image deblurring with Total-Variation (TV) prior

Poisson Inverse Problems with Maximum-Likelihood Expectation-Maximization (MLEM)

Poisson Inverse Problems with Maximum-Likelihood Expectation-Maximization (MLEM)

Image inpainting with wavelet prior

Image inpainting with wavelet prior

PnP with custom optimization algorithm (Primal-Dual Condat-Vu)

PnP with custom optimization algorithm (Primal-Dual Condat-Vu)

Plug-and-Play algorithm with Mirror Descent for Poisson noise inverse problems.

Plug-and-Play algorithm with Mirror Descent for Poisson noise inverse problems.

Vanilla PnP for computed tomography (CT).

Vanilla PnP for computed tomography (CT).