pytorch学习笔记-Transforms的使用

1.将image类型转换为tensorimage

from PIL import Image
# noinspection PyUnresolvedReferences
from torchvision import transforms
# python的用法->tensor数据类型
# 通过transform.ToTensor解决两个问题
# 绝对路径：D:\a\dataset\001.png
# 相对路径：dataset/001.png
img_path="D:\\a\\dataset\\001.png"
img=Image.open(img_path)
print(img)
# 1.transform如何使用
tensor_trans=transforms.ToTensor()
tensor_img=tensor_trans(img)
print(tensor_img)
# 2.tensor数据类型相较于普通类型有何区别
writer=SummaryWriter('logs')
writer.add_image("Tensor_img",tensor_img)
writer.close()

2.输出