From ea4bb5e83de6d931b552198b21616d85ae5897ba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sophia=20M=C3=A4dler?= <15019107+sophiamaedler@users.noreply.github.com> Date: Mon, 27 May 2024 18:49:40 +0200 Subject: [PATCH] specify figsize --- src/sparcscore/utils/vis.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/sparcscore/utils/vis.py b/src/sparcscore/utils/vis.py index c9c13a0..ffbcfa2 100644 --- a/src/sparcscore/utils/vis.py +++ b/src/sparcscore/utils/vis.py @@ -11,6 +11,7 @@ def plot_segmentation_mask( cmap_image="Greys_r", cmap_masks="jet", alpha=0.5, + figsize = (10, 10) ): """ Visualize the segmentation mask overlayed with a channel of the input image. @@ -52,7 +53,7 @@ def plot_segmentation_mask( segmentation = segmentation[mask_channel, selection[0], selection[1]] image = channels[image_channel, selection[0], selection[1]] - fig = plt.figure() + fig = plt.figure(figsize=figsize) plt.imshow(image, cmap=cmap_image) plt.imshow(segmentation, alpha=alpha, cmap=cmap_masks) plt.axis("off")