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[source]¶
在給定的 RGB 影像上繪製分割掩碼。影像值應為 uint8 型別，範圍在 [0, 255]；或 float 型別，範圍在 [0, 1]。
- 引數：
image (Tensor) – 形狀為 (3, H, W)，資料型別為 uint8 或 float 的張量。
masks (Tensor) – 形狀為 (num_masks, H, W) 或 (H, W)，資料型別為 bool 的張量。
alpha (float) – 介於 0 和 1 之間的浮點數，表示掩碼的透明度。0 表示完全透明，1 表示不透明。
colors (顏色 或 顏色列表, 可選) – 包含每個掩碼顏色的列表，或所有掩碼使用同一個顏色。顏色可以用 PIL 字串表示，例如“red”或“#FF00FF”；或用 RGB 元組表示，例如
(240, 10, 157)。預設情況下，會為每個掩碼生成隨機顏色。
- 返回值：
影像張量，在其上繪製了分割掩碼。
- 返回型別：
img (Tensor[C, H, W])
使用
draw_segmentation_masks的示例
