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