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

torchvision

Models, Datasets and Transformations for Images

Provides access to datasets, models and preprocessing facilities for deep learning with images. Integrates seamlessly with the 'torch' package and it's 'API' borrows heavily from 'PyTorch' vision package.

Functions (47)

base_loader

Base loader

cifar10_dataset

Cifar datasets

image_folder_dataset

Create an image folder dataset

kmnist_dataset

Kuzushiji-MNIST

magick_loader

Load an Image using ImageMagick

mnist_dataset

MNIST dataset

model_alexnet

AlexNet Model Architecture

model_resnet18

ResNet-18 Model Architecture

model_vgg

VGG implementation

tiny_imagenet_dataset

Tiny ImageNet dataset

transform_adjust_brightness

Adjust the brightness of an image

transform_adjust_contrast

Adjust the contrast of an image

transform_adjust_gamma

Adjust the gamma of an RGB image

transform_adjust_hue

Adjust the hue of an image

transform_adjust_saturation

Adjust the color saturation of an image

transform_affine

Apply affine transformation on an image keeping image center invariant

transform_center_crop

Crops the given image at the center

transform_color_jitter

Randomly change the brightness, contrast and saturation of an image

transform_convert_image_dtype

Convert a tensor image to the given dtype and scale the values accordingly

transform_crop

Crop the given image at specified location and output size

transform_five_crop

Crop image into four corners and a central crop

transform_grayscale

Convert image to grayscale

transform_hflip

Horizontally flip a PIL Image or Tensor

transform_linear_transformation

Transform a tensor image with a square transformation matrix and a mean_vector computed offline

transform_normalize

Normalize a tensor image with mean and standard deviation

transform_pad

Pad the given image on all sides with the given "pad" value

transform_perspective

Perspective transformation of an image

transform_random_affine

Random affine transformation of the image keeping center invariant

transform_random_apply

Apply a list of transformations randomly with a given probability

transform_random_choice

Apply single transformation randomly picked from a list

transform_random_crop

Crop the given image at a random location

transform_random_erasing

Randomly selects a rectangular region in an image and erases its pixel values

transform_random_grayscale

Randomly convert image to grayscale with a given probability

transform_random_horizontal_flip

Horizontally flip an image randomly with a given probability

transform_random_order

Apply a list of transformations in a random order

transform_random_perspective

Random perspective transformation of an image with a given probability

transform_random_resized_crop

Crop image to random size and aspect ratio

transform_random_rotation

Rotate the image by angle

transform_random_vertical_flip

Vertically flip an image randomly with a given probability

transform_resize

Resize the input image to the given size

transform_resized_crop

Crop an image and resize it to a desired size

transform_rgb_to_grayscale

Convert RGB Image Tensor to Grayscale

transform_rotate

Angular rotation of an image

transform_ten_crop

Crop an image and the flipped image each into four corners and a central crop

transform_to_tensor

Convert an image to a tensor

transform_vflip

Vertically flip a PIL Image or Tensor

vision_make_grid

A simplified version of torchvision.utils.make_grid

torchvision

Models, Datasets and Transformations for Images

v0.3.0

MIT + file LICENSE

Authors

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

Initial release

torchvision

Functions (47)

torchvision

We don't support your browser anymore