Table of Contents
Shortcuts

# draw_segmentation_masks

torchvision.utils.draw_segmentation_masks(image: Tensor, masks: Tensor, alpha: float = 0.8, colors: Optional[Union[List[Union[str, Tuple[int, int, int]]], str, Tuple[int, int, int]]] = None) → Tensor

  Draws segmentation masks on given RGB image. The values of the input image should be uint8 between 0 and 255.

  Parameters:

    - image (Tensor) – Tensor of shape (3, H, W) and dtype uint8.

    - masks (Tensor) – Tensor of shape (num_masks, H, W) or (H, W) and dtype bool.

    - alpha (float) – Float number between 0 and 1 denoting the transparency of the masks. 0 means full transparency, 1 means no transparency.

    - colors (color or list of colors, optional) – List containing the colors of the masks or single color for all masks. The color can be represented as PIL strings e.g. “red” or “#FF00FF”, or as RGB tuples e.g. `(240, 10, 157)`. By default, random colors are generated for each mask.

  Returns:

    Image Tensor, with segmentation masks drawn on top.

  Return type:

    img (Tensor[C, H, W])

  Examples using `draw_segmentation_masks`:

  Visualization utilities

  Visualization utilities

## Docs

Access comprehensive developer documentation for PyTorch

View Docs

## Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

## Resources

Find development resources and get your questions answered

View Resources

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.