译者: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)的序列,将会裁剪成矩形。
本文介绍torchvision库中transforms模块的功能,包括如何使用Compose组合多个图像变换,以及CenterCrop等常用变换的具体参数和用法。
2万+

被折叠的 条评论
为什么被折叠?



