不懂pytorch里transforms.Compose（）函数的同学看过来！

最新推荐文章于 2024-04-25 11:34:12 发布

阿二狗

最新推荐文章于 2024-04-25 11:34:12 发布

阅读量4.6k

点赞数 10

分类专栏： pytorch 笔记文章标签：深度学习 python pytorch

本文链接：https://blog.csdn.net/qq_43722906/article/details/115761415

版权

13 篇文章 0 订阅

订阅专栏

2 篇文章 0 订阅

订阅专栏

首先compose本身的英文含义有组成、排版、组合的意思。然后看到这里，我们大概可以猜出该函数的作用。（这其实也可以加深我们对该函数的记忆在这里插入图片描述
pytorch.torchvision.transform可以对PIL.Image进行各种变换

1.torchvision.transforms.CenterCrop(size)

将给定的PIL.Image进行中心切割，得到给定的size，size可以是tuple，(target_height, target_width)。size也可以是一个Integer，在这种情况下，切出来的图片的形状是正方形。

2.torchvision.transforms.RandomCrop(size, padding=0)

切割中心点的位置随机选取。size可以是tuple也可以是Integer。

3.torchvision.transforms.ToTensor

把一个取值范围是[0,255]的PIL.Image或者shape为(H,W,C)的numpy.ndarray，转换成形状为[C,H,W]，取值范围是[0,1.0]的torch.FloadTensor

等等，还有其他的一些函数，可以自己去查阅，我这里就不列举了

Compose就是对**transform.**的相关函数进行组合，使成一个代码块。

用法形式如下

transforms.Compose([
     transforms.CenterCrop(10),
     transforms.RandomCrop(size, padding=0)
     transforms.ToTensor(),
 ])

关注