当前位置: 代码迷 >> 综合 >> 深度学习框架_PyTorch_torchvision.transforms.functional.pad()函数
  详细解决方案

深度学习框架_PyTorch_torchvision.transforms.functional.pad()函数

热度:49   发布时间:2023-12-16 01:21:47.0
torchvision.transforms.functional.pad(img, padding, fill=0, padding_mode='constant')

这个函数用于用指定的填充模式和填充值填充PIL图像。

参数

  1. img(PIL图像)-- 要填充的图像。
  2. padding(int或tuple)-- 各边的填充值。如果指定为int,表示所有边都按照这个值进行填充。如果指定为长为2的元组,表示左右和上下边的填充值。如果指定为长为4的元组,分别表示左、上、右、下的填充宽度。
  3. fill – 要填充的像素值,默认是0.如果指定为长为3的元组,表示RGB三通道的填充值。这个选项在padding_mode是constant时有用。
  4. padding_mode – 填充类型,应当为:constant,edge,reflect或symmetric。默认是constant。
    constant:用常数填充,该常数值由fill指定。
    edge:用边上的值进行填充。
    reflect:以边为对称轴进行填充。(不重复边上的值)在reflect模式中,在两边分别用2个元素填充[1,2,3,4]将会得到[3,2,1,2,3,4,3,2]。
    symmetric:以边为对称轴进行填充。(重复边上的值)在symmetric模式中,在两边分别用2个元素填充[1,2,3,4]将会得到[2,1,1,2,3,4,4,3]。

返回值

填充后的图像。

返回类型

PIL图像。

  相关解决方案