- class torchvision.transforms.RandomRotation(degrees, interpolation=InterpolationMode.NEAREST, expand=False, center=None, fill=0)[source]¶
Rotate the image by angle. If the image is torch Tensor, it is expected to have […, H, W] shape, where … means an arbitrary number of leading dimensions.
degrees (sequence or number) – Range of degrees to select from. If degrees is a number instead of sequence like (min, max), the range of degrees will be (-degrees, +degrees).
interpolation (InterpolationMode) – Desired interpolation enum defined by
torchvision.transforms.InterpolationMode. Default is
InterpolationMode.NEAREST. If input is Tensor, only
InterpolationMode.BILINEARare supported. The corresponding Pillow integer constants, e.g.
PIL.Image.BILINEARare accepted as well.
expand (bool, optional) – Optional expansion flag. If true, expands the output to make it large enough to hold the entire rotated image. If false or omitted, make the output image the same size as the input image. Note that the expand flag assumes rotation around the center and no translation.
center (sequence, optional) – Optional center of rotation, (x, y). Origin is the upper left corner. Default is the center of the image.
fill (sequence or number) – Pixel fill value for the area outside the rotated image. Default is
0. If given a number, the value is used for all bands respectively.
Getting started with transforms v2Getting started with transforms v2Illustration of transforms
img (PIL Image or Tensor) – Image to be rotated.
- Return type:
PIL Image or Tensor