Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

transform_to_tensor

Convert an image to a tensor


Description

Converts a Magick Image or array (H x W x C) in the range [0, 255] to a torch_tensor of shape (C x H x W) in the range [0.0, 1.0]. In the other cases, tensors are returned without scaling.

Usage

transform_to_tensor(img)

Arguments

img

A magick-image, array or torch_tensor.

Note

Because the input image is scaled to [0.0, 1.0], this transformation should not be used when transforming target image masks.

See Also


torchvision

Models, Datasets and Transformations for Images

v0.3.0
MIT + file LICENSE
Authors
Daniel Falbel [aut, cre], RStudio [cph]
Initial release

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.