PyTorch 1.0 中文文档:torchvision.transforms

译者:BXuan694

transforms包含了一些常用的图像变换,这些变换能够用Compose串联组合起来。另外,torchvision提供了torchvision.transforms.functional模块。functional可以提供了一些更加精细的变换,用于搭建复杂的变换流水线(例如分割任务)。

class torchvision.transforms.Compose(transforms)

用于把一系列变换组合到一起。

参数:

  • transforms(list或Transform对象)- 一系列需要进行组合的变换。

示例:

>>> transforms.Compose([
>>>     transforms.CenterCrop(10),
>>>     transforms.ToTensor(),
>>> ])

对PIL图片的变换

class torchvision.transforms.CenterCrop(size)

在中心处裁剪PIL图片。

参数:

  • size序列 int)– 需要裁剪出的形状。如果size是int,将会裁剪成正方形;如果是形如(h, w)的序列,将会裁剪成矩形。

阅读全文/改进本文

猜你喜欢

转载自www.cnblogs.com/wizardforcel/p/10492701.html