transform就是一个工具箱,里面有很多工具,可以输出我们所需要的图片
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
from PIL import Image
# 转tensor数据类型
# 通过transforms.ToTensor
# 1.transforms该如何使用
# 2.Tensor数据类型有什么区别
image_path = 'hymenoptera_data/train/ants/0013035.jpg'
image_PIL=Image.open(image_path)
writer=SummaryWriter("logs")
# print(image_PIL)
# 引入对象
tensor_trans = transforms.ToTensor()
tensor_image = tensor_trans(image_PIL)
print(tensor_image)
writer.add_image('tensorimage',tensor_image)
# 别忘了writer.close
writer.close()
tensor数据类型包含了神经网络 所需要的很多参数