译者: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)的序列,将会裁剪成矩形。
原文地址:https://www.cnblogs.com/wizardforcel/p/10492701.html
时间: 2024-10-13 15:23:22