mipcandy.data.visualization#

Module Contents#

Functions#

visualize2d

_visualize3d_with_pyvista

visualize3d

overlay

param image:

base image

param label:

label (class ids) to overlay on top of image

param max_label_opacity:

maximum opacity of the label

param label_colorizer:

colorizer for the label, defaults to ColorizeLabel()

Data#

API#

mipcandy.data.visualization.visualize2d(image: torch.Tensor, *, title: str | None = None, cmap: str | None = None, is_label: bool = False, blocking: bool = False, screenshot_as: str | os.PathLike[str] | None = None) None[source]#
mipcandy.data.visualization._visualize3d_with_pyvista(image: numpy.ndarray, title: str | None, cmap: str, screenshot_as: str | os.PathLike[str] | None) None[source]#
mipcandy.data.visualization.__LABEL_COLORMAP: list[str] = ['#ffffff', '#2e4057', '#7a0f1c', '#004f4f', '#9a7b00', '#2c2f38', '#5c136f', '#113f2e', '#8a3b12', ...#
mipcandy.data.visualization.visualize3d(image: torch.Tensor, *, title: str | None = None, cmap: str | list[str] | None = None, max_volume: int = 1000000.0, is_label: bool = False, backend: Literal[auto, matplotlib, pyvista] = 'auto', blocking: bool = False, screenshot_as: str | os.PathLike[str] | None = None) None[source]#
mipcandy.data.visualization.overlay(image: torch.Tensor, label: torch.Tensor, *, max_label_opacity: float = 0.5, label_colorizer: mipcandy.common.ColorizeLabel | None = ColorizeLabel(batch=False)) torch.Tensor[source]#
Parameters:
  • image – base image

  • label – label (class ids) to overlay on top of image

  • max_label_opacity – maximum opacity of the label

  • label_colorizer – colorizer for the label, defaults to ColorizeLabel()