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, 可選) – 網格中每行顯示的影像數量。最終網格尺寸為
(B / nrow, nrow)。預設值:8。padding (int, 可選) – 填充量。預設值:
2。normalize (bool, 可選) – 如果為 True，則根據
value_range指定的最小值和最大值，將影像值歸一化到 (0, 1) 範圍。預設值:False。value_range (tuple, 可選) – (min, max) 元組，其中 min 和 max 是數字，這些數字用於對影像進行歸一化。預設情況下，min 和 max 是從 tensor 中計算得出的。
scale_each (bool, 可選) – 如果為
True，則分別縮放影像批次中的每張影像，而不是使用所有影像的 (min, max) 進行縮放。預設值:False。pad_value (float, 可選) – 填充畫素的值。預設值:
0。
- 返回:
包含影像網格的 tensor。
- 返回型別:
grid (Tensor)
使用
make_grid的示例
