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 是一個數字，則將在 (-shear, +shear) 的範圍內應用平行於 x 軸的剪下。否則，如果 shear 是 2 個值的序列，則將在 (shear[0], shear[1]) 的範圍內應用平行於 x 軸的剪下。否則，如果 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]]

獲取仿射變換的引數

返回:

要傳遞給仿射變換的引數