RandomAffine

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

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

参数:

• 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)，则 scale 从范围 a <= scale <= b 中随机采样。默认情况下将保持原始缩放。

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

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

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

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

使用 RandomAffine 的示例

转换图示

转换图示

forward(img)

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

返回:

仿射变换后的图像。

返回类型:

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

获取仿射变换的参数

返回:

要传递给仿射变换的参数