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/
标签:PIL,torchvision,Image,用法,pytorch,transforms,操作 来源: https://www.cnblogs.com/h694879357/p/15985491.html