draw_segmentation_masks — Torchvision main documentation
https://pytorch.org/.../torchvision.utils.draw_segmentation_masks.htmldraw_segmentation_masks¶ torchvision.utils. draw_segmentation_masks (image: torch.Tensor, masks: torch.Tensor, alpha: float = 0.8, colors: Optional [Union [List [Union [str, Tuple [int, int, int]]], str, Tuple [int, int, int]]] = None) → torch.Tensor [source] ¶ Draws segmentation masks on given RGB image. The values of the input image should be uint8 between 0 and 255.