RandomAffine¶
-
class
torchvision.transforms.
RandomAffine
(degrees, translate=None, scale=None, shear=None, interpolation=<InterpolationMode.NEAREST: 'nearest'>, fill=0, fillcolor=None, resample=None, center=None)[source]¶ Random affine transformation of the image keeping center invariant. If the image is torch Tensor, it is expected to have […, H, W] shape, where … means an arbitrary number of leading dimensions.
- Parameters
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). Set to 0 to deactivate rotations.
translate (tuple, optional) – tuple of maximum absolute fraction for horizontal and vertical translations. For example translate=(a, b), then horizontal shift is randomly sampled in the range -img_width * a < dx < img_width * a and vertical shift is randomly sampled in the range -img_height * b < dy < img_height * b. Will not translate by default.
scale (tuple, optional) – scaling factor interval, e.g (a, b), then scale is randomly sampled from the range a <= scale <= b. Will keep original scale by default.
shear (sequence or number, optional) – Range of degrees to select from. If shear is a number, a shear parallel to the x axis in the range (-shear, +shear) will be applied. Else if shear is a sequence of 2 values a shear parallel to the x axis in the range (shear[0], shear[1]) will be applied. Else if shear is a sequence of 4 values, a x-axis shear in (shear[0], shear[1]) and y-axis shear in (shear[2], shear[3]) will be applied. Will not apply shear by default.
interpolation (InterpolationMode) – Desired interpolation enum defined by
torchvision.transforms.InterpolationMode
. Default isInterpolationMode.NEAREST
. If input is Tensor, onlyInterpolationMode.NEAREST
,InterpolationMode.BILINEAR
are supported. For backward compatibility integer values (e.g.PIL.Image.NEAREST
) are still acceptable.fill (sequence or number) – Pixel fill value for the area outside the transformed image. Default is
0
. If given a number, the value is used for all bands respectively.fillcolor (sequence or number, optional) –
Warning
This parameter was deprecated in
0.12
and will be removed in0.14
. Please usefill
instead.resample (int, optional) –
Warning
This parameter was deprecated in
0.12
and will be removed in0.14
. Please useinterpolation
instead.center (sequence, optional) – Optional center of rotation, (x, y). Origin is the upper left corner. Default is the center of the image.
Examples using
RandomAffine
:-
forward
(img)[source]¶ img (PIL Image or Tensor): Image to be transformed.
- Returns
Affine transformed image.
- Return type
PIL Image or Tensor
-
static
get_params
(degrees: List[float], translate: Optional[List[float]], scale_ranges: Optional[List[float]], shears: Optional[List[float]], img_size: List[int]) → Tuple[float, Tuple[int, int], float, Tuple[float, float]][source]¶ Get parameters for affine transformation
- Returns
params to be passed to the affine transformation