合集请看：pyTorch入门合集
参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click

transforms在python中如何使用

from PIL import Image  
from torchvision import transforms  

# 读取图片  
img_path = "dataset/train/ants_image/0013035.jpg"  
img = Image.open(img_path)  
print(img)  
# transforms的用法  
tensor_trans = transforms.ToTensor()  
tensor_img = tensor_trans(img)  
print(tensor_img)

为什么我们需要Tensor数据类型

Tensor类比python自带的Image类多携带了许多参数，可以理解为是为了方便神经网络计算的一种特殊数据类型