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

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)