torchvision: vision_make_grid – R documentation

Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!

vision_make_grid

A simplified version of torchvision.utils.make_grid

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).

vision_make_grid(
  tensor,
  scale = TRUE,
  num_rows = 8,
  padding = 2,
  pad_value = 0
)

`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.

Models, Datasets and Transformations for Images

v0.3.0

MIT + file LICENSE

Authors

Daniel Falbel [aut, cre], RStudio [cph]

Initial release