码迷,mamicode.com
首页 > 其他好文 > 详细

PyTorch 1.0 中文文档:torchvision.transforms

时间:2019-03-07 22:10:34      阅读:816      评论:0      收藏:0      [点我收藏+]

标签:forms   改进   module   class   1.0   图片   git   transform   .com   

译者: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)的序列,将会裁剪成矩形。

阅读全文/改进本文

PyTorch 1.0 中文文档:torchvision.transforms

标签:forms   改进   module   class   1.0   图片   git   transform   .com   

原文地址:https://www.cnblogs.com/wizardforcel/p/10492701.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!