这里有几个要注意的点,首先是关于控制台中打开tensorboard的链接,有时候会发现链接点开以后找不到文件,这是因为控制台中的路径不对,你要在控制台中cd到你生成图片的文件夹的上一级,才行,其次这里的normalize中的参数,均值和标准差是我上一篇文章的代码的输出结果
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
from PIL import Image
image_path="E:/Jupyter/pycharm/data/train/ants_image/0013035.jpg"
image=Image.open(image_path)#打开蚂蚁文件
t=transforms.ToTensor()#实例化Totensor
image_tensor=t(image)#将照片转化为tensor类型
write=SummaryWriter("logs")#创建问价夹
write.add_image("not change",image_tensor)#添加照片
tensor_nomal=transforms.Normalize([0.3,0.5,0.8],[0.08,0.11,0.19])#实例化normalize,输入照片均值和标准差
image_change=tensor_nomal(image_tensor)#将照片通过设置好的normalize进行转化
write.add_image("change",image_change)#添加转化后的照片
write.close()#关闭SummaryWriter
输出结果