make_grid¶
- torchvision.utils.make_grid(tensor: Union[Tensor, List[Tensor]], nrow: int = 8, padding: int = 2, normalize: bool = False, value_range: Optional[Tuple[int, int]] = None, scale_each: bool = False, pad_value: float = 0.0) Tensor[source]¶
將影像製作成網格。
- 參數:
tensor (Tensor 或 list) – 形狀為 (B x C x H x W) 的 4D mini-batch Tensor 或所有大小相同的影像列表。
nrow (int, optional) – 網格中每行顯示的影像數量。最終網格大小為
(B / nrow, nrow)。預設值：8。padding (int, optional) – 填充量。預設值：
2。normalize (bool, optional) – 如果為 True，則透過
value_range指定的最小值和最大值，將影像移動到範圍 (0, 1)。預設值：False。value_range (tuple, optional) – tuple (min, max)，其中 min 和 max 是數字，然後這些數字用於正規化影像。預設情況下，min 和 max 從 tensor 計算得出。
scale_each (bool, optional) – 如果為
True，則分別縮放影像批次中的每個影像，而不是縮放所有影像的 (min, max)。預設值：False。pad_value (float, optional) – 填充像素的值。預設值：
0。
- 回傳值:
包含影像網格的 tensor。
- 回傳型別:
grid (Tensor)
使用
make_grid的範例
