PyTorch库 | (二) torchvision.transforms

torchvision.transforms

• PyTorch中文文档：pytorch torchvision transform
• PyTorch源码解读（二）torchvision.transforms

torchvision.transforms这个包中包含resize、crop等常见的data augmentation操作，基本上PyTorch中的data augmentation操作都可以通过该接口实现。该包主要包含两个脚本：transformas.py和functional.py，前者定义了各种data augmentation的类，在每个类中通过调用functional.py中对应的函数完成data augmentation操作。

该包主要包含五类变换：

1. Transforms on PIL Image（对PIL.Image进行变换）
2. Transforms on torch.Tensor（对Tensor进行变换）
3. Conversion Transforms
4. Generic Transforms（通用变换）
5. Functional Transforms

1. Transforms on PIL Image

1. class torchvision.transforms.Compose(transforms) ：将多个transform组合起来使用。
2. torchvision.transforms.RandomOrder（transforms） #以随机顺序应用转换列表。
3. torchvision.transforms.RandomChoice(transforms)：应用从列表中随机挑选的单个转换。
4. 大小
5. torchvision.transforms.Resize(size，in