torchvision.transforms用法


为了方便进行数据的操作,pytorch团队提供了一个torchvision.transforms包,我们可以用transforms进行以下操作:

  • PIL.Image/numpy.ndarray与Tensor的相互转化;
  • 归一化;
  • 对PIL.Image进行裁剪、缩放等操作。

 

通常,在使用torchvision.transforms,我们通常使用transforms.Compose将transforms组合在一起。

详细看文档:https://pytorch-cn.readthedocs.io/zh/latest/torchvision/torchvision-transform/