vision_make_grid: A simplified version of torchvision.utils.make_grid

Description Usage Arguments

View source: R/vision_utils.R

Description

Arranges a batch of (image) tensors in a grid, with optional padding between images. Expects a 4d mini-batch tensor of shape (B x C x H x W).

Usage

1
2
3
4
5
6
7
vision_make_grid(
  tensor,
  scale = TRUE,
  num_rows = 8,
  padding = 2,
  pad_value = 0
)

Arguments

tensor

tensor to arrange in grid.

scale

whether to normalize (min-max-scale) the input tensor.

num_rows

number of rows making up the grid (default 8).

padding

amount of padding between batch images (default 2).

pad_value

pixel value to use for padding.


torchvision documentation built on Aug. 17, 2021, 5:06 p.m.