Pytorch学习日记——Transforms使用（一）

学习视频——B站【小土堆】

 本节内容是将图片转化为tensor形式

from PIL import Image
from torchvision import transforms


# 绝对路径 C:\py\PycharmProjects\pythonproject1\dataset\train\erciyuan_image\407-22641.jpg
# 相对路径 dataset/train/erciyuan_image/407-22641.jpg
img_path ="dataset/train/erciyuan_image/407-22641.jpg"
img = Image.open(img_path)
print(img)

# transforms该如何使用(python)
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)


print(tensor_img)

输出