torchvision.transform
变换是常用的图像变换。它们可以用Compose
连接在一起。
class torchvision.transforms.Compose(transforms)
将多个transform组合起来使用。
transforms: 由transform构成的列表. 例子:
transforms.Compose([
transforms.CenterCrop(10),
transforms.ToTensor(),
])
对PIL.Image进行变换
class torchvision.transforms.Scale(size, interpolation=2)
按照规定的尺寸重新调节PIL.Image
。
参数说明:
- size (sequence or int) - 期望输出尺寸。如果size是一个像(w, h)的序列,输出大小将按照w,h匹配到。如果大小是int,则图像将匹配到这个数字。例如,如果原图的
height>width
,那么改变大小后的图片大小是(size*height/width, size)
。 - interpolation (int, optional) -需要添加值。默认的是
PIL.Image.BILINEAR
class torchvision.transforms.CenterCrop(size)
将给定的PIL.Image进行中心切割,得到给定的size,size可以是tuple,(target_height, target_width)。size也可以是一个Integer,在这种情况下,切出来的图片的形状是正方形。
class torchvision.transforms.RandomCrop(size, padding=0)
切割中心点的位置随机选取。size可以是tuple也可以是Integer。
class torchvision.transforms.RandomHorizontalFlip
随机水平翻转给定的PIL.Image,概率为0.5。即:一半的概率翻转,一半的概率不翻转。
class torchvision.transforms.RandomSizedCrop(size, interpolation=2)
先将给定的PIL.Image随机切,然后再resize成给定的size大小。
class torchvision.transforms.Pad(padding, fill=0)
将给定的PIL.Image的所有边用给定的pad value填充。 padding:要填充多少像素 fill:用什么值填充
对Tensor进行变换
class torchvision.transforms.Normalize(mean, std)
给定均值:(R,G,B) 方差:(R,G,B),将会把Tensor正则化。即:Normalized_image=(image-mean)/std。
参数说明:
- mean (sequence) – 序列R, G, B的均值。
- std (sequence) – 序列 R, G, B 的平均标准偏差.
**call(tensor)** 参数: tensor (Tensor) – 规范化的大小(c,h,w)的张量图像. 返回结果: 规范化的图片. 返回样式: Tensor张量
对Conversion进行变换
class torchvision.transforms.ToTensor
把一个取值范围是[0,255]的PIL.Image或者shape为(H,W,C)的numpy.ndarray,转换成形状为[C,H,W],取值范围是[0,1.0]的torch.FloadTensor
**call(pic)**
- 参数: pic (PIL.Image or numpy.ndarray) – 图片转换为张量.
- 返回结果: 转换后的图像。
- 返回样式: Tensor张量
class torchvision.transforms.ToPILImage
将shape为(C,H,W)的Tensor或shape为(H,W,C)的numpy.ndarray转换成PIL.Image,值不变。
**call(pic)**
- 参数: pic (Tensor or numpy.ndarray) – 图像转换为pil.image。
- 返回结果: 图像转换为PIL.Image.
- 返回样式: PIL.Image
通用变换
class torchvision.transforms.Lambda(lambd)
使用lambd作为转换器。
参数说明: lambd (function) – Lambda/function 用于转换.
译者署名
用户名 | 头像 | 职能 | 签名 |
---|---|---|---|
Song | 翻译 | 人生总要追求点什么 |