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 影像上繪製分割掩碼。影像值應為 [0, 255] 範圍內的 uint8 或 [0, 1] 範圍內的 float。
- 引數:
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的示例
