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:

Radio interferometric imaging with deepinverse

Radio interferometric imaging with deepinverse

Image deblurring with Total-Variation (TV) prior

Image deblurring with Total-Variation (TV) prior

Image inpainting with wavelet prior

Image inpainting with wavelet prior

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).

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

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