填充

class torchvision.transforms.Pad(padding, fill=0, padding_mode='constant')

使用給定的“填充”值填充給定影像的所有邊。如果影像是 torch Tensor，則其形狀應為 […, H, W]，其中 … 表示對於 reflect 和 symmetric 模式，最多有 2 個前導維度；對於 edge 模式，最多有 3 個前導維度；對於 constant 模式，前導維度數量不限

引數:

• padding (int 或 sequence) –

  每個邊界上的填充量。如果提供一個單獨的 int 值，則用於填充所有邊界。如果提供一個長度為 2 的 sequence，則分別表示左右和頂部/底部邊界的填充量。如果提供一個長度為 4 的 sequence，則分別表示左、頂、右和底邊界的填充量。

  注意

  在 torchscript 模式下，不支援單個 int 值的填充，請使用長度為 1 的 sequence：[padding, ]。

• fill (number 或 tuple) – 對於 constant 填充的畫素填充值。預設為 0。如果是一個長度為 3 的 tuple，則分別用於填充 R、G、B 通道。此值僅在 padding_mode 為 constant 時使用。對於 torch Tensor，僅支援 number 型別。對於 PIL Image，僅支援 int 或 tuple 值。

• padding_mode (str) –

  填充型別。應為：constant, edge, reflect 或 symmetric。預設為 constant。

  • constant: 使用一個常量值進行填充，此值透過 fill 指定

  • edge: 使用影像邊緣的最後一個值進行填充。如果輸入是 5D torch Tensor，將填充最後 3 個維度而不是最後 2 個維度

  • reflect: 使用影像的反射進行填充，但不重複邊緣的最後一個值。例如，在 reflect 模式下，在 [1, 2, 3, 4] 的兩側填充 2 個元素後，結果將是 [3, 2, 1, 2, 3, 4, 3, 2]

  • symmetric: 使用影像的反射進行填充，並重復邊緣的最後一個值。例如，在 symmetric 模式下，在 [1, 2, 3, 4] 的兩側填充 2 個元素後，結果將是 [2, 1, 1, 2, 3, 4, 4, 3]

使用 Pad 的示例

變換圖示

變換圖示

forward(img)

引數:

img (PIL Image 或 Tensor) – 待填充的影像。

返回:

填充後的影像。

返回型別:

PIL Image 或 Tensor