specify figsize

MannLabs · May 27, 2024 · ea4bb5e · ea4bb5e
1 parent 5747088
commit ea4bb5e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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")