Pad¶

class torchvision.transforms.Pad(padding, fill=0, padding_mode='constant')[原始碼]¶

在給定影像的四周用給定的“pad”值進行填充。如果影像是 torch Tensor，則期望其形狀為 […, H, W]，其中 … 表示模式 reflect 和 symmetric 最多可以有 2 個前導維度，模式 edge 最多可以有 3 個前導維度，而模式 constant 可以有任意數量的前導維度。

引數:

padding (int 或 sequence) –
每個邊框的填充。如果提供單個整數，則用於填充所有邊框。如果提供長度為2的序列，則分別表示左右和上下邊框的填充。如果提供長度為4的序列，則分別表示左、上、右和下邊框的填充。

注意

在 torchscript 模式下，不支援將 padding 作為單個 int，請使用長度為 1 的序列：[padding, ]。
fill (number 或 tuple) – 常量填充的畫素填充值。預設為 0。如果為長度為 3 的元組，則分別用於填充 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]