Pad¶

class torchvision.transforms.v2.Pad(padding: Union[int, Sequence[int]], fill: Union[int, float, Sequence[int], Sequence[float], None, dict[Union[type, str], Union[int, float, collections.abc.Sequence[int], collections.abc.Sequence[float], NoneType]] = 0, padding_mode: Literal['constant', 'edge', 'reflect', 'symmetric'] = 'constant')[原始碼]¶

使用給定的“填充”值在所有側面填充輸入。

如果輸入是 torch.Tensor 或 TVTensor (例如 Image, Video, BoundingBoxes 等) 它可能具有任意數量的前導批處理維度。例如，影像可以具有 [..., C, H, W] 形狀。邊界框可以具有 [..., 4] 形狀。

引數:

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

注意

在 torchscript 模式下，不支援將 padding 作為單個 int，請使用長度為 1 的序列：[padding, ]。
fill (數字或元組或字典, 可選) – 當 padding_mode 為 constant 時使用的畫素填充值。預設為 0。如果是一個長度為 3 的元組，它分別用於填充 R、G、B 通道。填充值也可以是一個對映資料型別到填充值的字典，例如 fill={tv_tensors.Image: 127, tv_tensors.Mask: 0}，其中 Image 將被填充為 127，Mask 將被填充為 0。
padding_mode (字串, 可選) –
填充型別。應為：constant、edge、reflect 或 symmetric。預設為“constant”。
- constant：用常數值填充，該值透過 fill 指定
- edge：使用影像邊緣的最後一個值進行填充。
- 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]