Pytorch学习（3）transforms的使用

transforms是pytorch中的图像处理包,一般用Compose把多个处理整合到一起

使用transforms的一般步骤是：

1.创建具体的工具：tool = transforms.ToTensor()

2.使用工具：result = tool（input）

from PIL import Image
from torchvision import transforms

img_path = "dataset/train/ants_image/0013035.jpg"
img = Image.open(img_path)
print(img)

tensor_trans = transforms.ToTensor()   #类实例化（创建具体的工具）
tensor_img = tensor_trans(img)

print(tensor_img)

可以看到图片已经被转化为Tensor类型

 为什么需要转换为Tensor数据类型？

因为训练神经网络的 反向传播，计算梯度等都需要Tensor类型的数据。