RandomAffine¶

class torchvision.transforms.v2.RandomAffine(degrees: Union[Number, Sequence], translate: Optional[Sequence[float]] = None, scale: Optional[Sequence]] = None, shear: Optional[Union[int, float, Sequence]]] = None, interpolation: Union[InterpolationMode, int] = InterpolationMode.NEAREST, fill: Union[int, float, Sequence], Sequence], None, Dict[Union[Type, str], Optional[Union[int, float, Sequence], Sequence[float]]]]] = 0, center: Optional[List]float]] = None)[source]¶

随机仿射变换输入，保持中心不变。

如果输入是 torch.Tensor 或 TVTensor (例如 Image, Video, BoundingBoxes 等)，它可以具有任意数量的前导批次维度。例如，图像可以具有 [..., C, H, W] 形状。边界框可以具有 [..., 4] 形状。

参数:

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 是一个数字，则将应用与 x 轴平行的 (-shear, +shear) 范围内的剪切。否则，如果 shear 是 2 个值的序列，则将应用与 x 轴平行的 (shear[0], shear[1]) 范围内的剪切。否则，如果 shear 是 4 个值的序列，则将应用 (shear[0], shear[1]) 中的 x 轴剪切和 (shear[2], shear[3]) 中的 y 轴剪切。默认情况下不应用剪切。
interpolation (InterpolationMode, optional) – 期望的插值枚举，由 torchvision.transforms.InterpolationMode 定义。默认为 InterpolationMode.NEAREST。如果输入是 Tensor，则仅支持 InterpolationMode.NEAREST, InterpolationMode.BILINEAR。也接受相应的 Pillow 整数常量，例如 PIL.Image.BILINEAR。
fill (number 或 tuple 或 dict, optional) – 当 padding_mode 为 constant 时使用的像素填充值。默认为 0。如果长度为 3 的元组，则分别用于填充 R、G、B 通道。填充值也可以是将数据类型映射到填充值的字典，例如 fill={tv_tensors.Image: 127, tv_tensors.Mask: 0}，其中 Image 将填充为 127，Mask 将填充为 0。
center (sequence, optional) – 可选的旋转中心，(x, y)。原点是左上角。默认为图像的中心。