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/

posted @ 2022-03-09 15:48  Tomorrow1126  阅读(140)  评论(0)  编辑  收藏  举报