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[源代码]¶
在给定的 RGB 图像上绘制分割掩码。图像值应为 uint8（范围 [0, 255]）或 float（范围 [0, 1]）。
- 参数:
image (Tensor) – 形状为 (3, H, W) 且 dtype 为 uint8 或 float 的 Tensor。
masks (Tensor) – 形状为 (num_masks, H, W) 或 (H, W) 且 dtype 为 bool 的 Tensor。
alpha (float) – 一个介于 0 和 1 之间的浮点数，表示掩码的透明度。0 表示完全透明，1 表示不透明。
colors (颜色 或 颜色列表, 可选) – 包含掩码颜色的列表，或用于所有掩码的单个颜色。颜色可以表示为 PIL 字符串（例如，“red”或“#FF00FF”），或表示为 RGB 元组（例如，
(240, 10, 157)）。默认情况下，会为每个掩码生成随机颜色。
- 返回:
图像 Tensor，带有绘制的分割掩码。
- 返回类型:
img (Tensor[C, H, W])
使用
draw_segmentation_masks的示例
