wandb_figure

wandb_figure

Helper for converting matplotlib figures into wandb images.

figure_to_wandb_image

figure_to_wandb_image(fig, dpi=100)

Convert matplotlib figure to WandB image.

Parameters:

Name	Type	Description	Default
fig	Figure	Matplotlib figure	required
dpi	int	Resolution for the rasterized PNG buffer	100

Returns:

Type	Description
Image	WandB image object

Source code in src/versatil/training/callbacks/wandb_figure.py

def figure_to_wandb_image(fig: plt.Figure, dpi: int = 100) -> wandb.Image:
    """Convert matplotlib figure to WandB image.

    Args:
        fig: Matplotlib figure
        dpi: Resolution for the rasterized PNG buffer

    Returns:
        WandB image object
    """
    buf = io.BytesIO()
    fig.savefig(buf, format="png", dpi=dpi, bbox_inches="tight")
    buf.seek(0)
    pil_img = Image.open(buf)
    return wandb.Image(pil_img)