随机仿射

class torchvision.transforms.RandomAffine(degrees, translate=None, scale=None, shear=None, interpolation=InterpolationMode.NEAREST, fill=0, center=None)

对图像进行随机仿射变换，保持中心不变。 如果图像是torch张量，则期望其具有[…, H, W]形状，其中…表示任意数量的前导维度。

Parameters:

• degrees (sequence 或 number) – 选择的角度范围。 如果 degrees 是一个数字而不是像 (min, max) 这样的序列，则角度范围 将为 (-degrees, +degrees)。设置为 0 以停用旋转。

• translate (tuple, optional) – 水平和垂直平移的最大绝对比例的元组。例如 translate=(a, b)，则水平位移在 -img_width * a < dx < img_width * a 范围内随机采样，垂直位移在 -img_height * b < dy < img_height * b 范围内随机采样。默认情况下不会进行平移。

• scale (tuple, optional) – 缩放因子区间，例如 (a, b)，然后从范围 a <= scale <= b 中随机采样。默认情况下将保持原始比例。

• shear (sequence 或 number, optional) – 选择的范围角度。 如果 shear 是一个数字，将在 (-shear, +shear) 范围内应用平行于 x 轴的剪切。 否则，如果 shear 是 2 个值的序列，将在 (shear[0], shear[1]) 范围内应用平行于 x 轴的剪切。 否则，如果 shear 是 4 个值的序列，将在 (shear[0], shear[1]) 范围内应用 x 轴剪切，并在 (shear[2], shear[3]) 范围内应用 y 轴剪切。 默认情况下不会应用剪切。

• 插值 (InterpolationMode) – 由torchvision.transforms.InterpolationMode定义的所需插值枚举。默认值为InterpolationMode.NEAREST。 如果输入是张量，则仅支持InterpolationMode.NEAREST和InterpolationMode.BILINEAR。 也接受相应的Pillow整数常量，例如PIL.Image.BILINEAR。

• fill (sequence 或 number) – 变换图像外部区域的像素填充值。默认值为 0。如果给定一个数字，则该值将分别用于所有波段。

• center (sequence, optional) – 可选的旋转中心，(x, y)。原点是左上角。 默认是图像的中心。

使用 RandomAffine 的示例：

变换的图示

Illustration of transforms

forward(img)

img (PIL Image 或 Tensor): 要转换的图像。

Returns:

仿射变换后的图像。

Return type:

PIL 图像或张量

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]]

获取仿射变换的参数

Returns:

传递给仿射变换的参数