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类型的数据。