GeometryΒΆ

CenterCrop(size)

Crop the input at the center.

ElasticTransform([alpha, sigma, ...])

Transform the input with elastic transformations.

FiveCrop(size)

Crop the image or video into four corners and the central crop.

Pad(padding[, fill, padding_mode])

Pad the input on all sides with the given "pad" value.

RandomAffine(degrees[, translate, scale, ...])

Random affine transformation the input keeping center invariant.

RandomCrop(size[, padding, pad_if_needed, ...])

Crop the input at a random location.

RandomHorizontalFlip([p, batch_inplace, ...])

Horizontally flip the input with a given probability.

RandomIoUCrop([min_scale, max_scale, ...])

Random IoU crop transformation from "SSD: Single Shot MultiBox Detector".

RandomPerspective([distortion_scale, p, ...])

Perform a random perspective transformation of the input with a given probability.

RandomResize(min_size, max_size[, ...])

Randomly resize the input.

RandomResizedCrop(size[, scale, ratio, ...])

Crop a random portion of the input and resize it to a given size.

RandomRotation(degrees[, interpolation, ...])

Rotate the input by angle.

RandomShortestSize(min_size[, max_size, ...])

Randomly resize the input.

RandomVerticalFlip([p, batch_inplace, ...])

Vertically flip the input with a given probability.

RandomZoomOut([fill, side_range, p])

Resize(size[, interpolation, max_size, ...])

Resize the input to the given size.

ScaleJitter(target_size[, scale_range, ...])

Perform Large Scale Jitter on the input according to "Simple Copy-Paste is a Strong Data Augmentation Method for Instance Segmentation".

TenCrop(size[, vertical_flip])

Crop the image or video into four corners and the central crop plus the flipped version of these (horizontal flipping is used by default).