1.安装tensoboard
在激活对应环境的conda里面输入:(需要提前安装好pytorch,安装方法:https://blog.csdn.net/m0_53971297/article/details/139231657?spm=1001.2014.3001.5501)
pip install tensorboard
2.绘制图像
创建一共y=x的图像:
-
from torch.utils.tensorboard import SummaryWriter writer =SummaryWriter("logs") #writer.add_image() #writer.add_scalar( ) #y=x for i in range(100): writer.add_scalar("y=x",i,i) writer.close()
3.通过tensorboard查看图片
再tensorboard里面查看图片:
-
from torch.utils.tensorboard import SummaryWriter import numpy as np from PIL import Image writer =SummaryWriter("logs") image_path =""#放图片的路径 img_PIL =Image.open(image_path) img_array = np.array(img_PIL) print(img_array.shape) writer.add_image("test",img_array,1) #writer.add_scalar( ) #y=x for i in range(100): writer.add_scalar("y=2x",2*i,i) writer.close()
如果image_path 报错:
1,更改为绝对路径;
2.在一个新的地址里面打开这个dataset;
3.用/不用\。
writer.add_image("test",img_array,1,dataformats='HWC') 更改“”里面的名字,变成不同命名种类;更改里面的数字,可添加多个图片到不同step;通过print(img_array.shape)查看图片形式 如果不算3HW就需要添加dataformats形式
4.打开tensorboard
运行上面的代码后,在terminal 里面输入:
tensorboard --logdir=“logs的绝对路径“
下面弹出tensorboard链接,直接=logs可能打开没有图像,变成“logs绝对路径,不能有中文“
更改端口方法:tensorboard --logdir=“logs的绝对路径“ –port=6007
如果出现图像混乱,可以删去logs文件下所有文件重新跑