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(type(img_array)) print(img_array.shape) ''' 名称 必须是Numpy形式的图片 第三个参数:代表步骤 图片的形式 ''' writer.add_image("train", img_array, 1, dataformats='HWC') # y = 2x # 画坐标图 # for i in range(100): # writer.add_scalar("y=2x", 3*i, i) writer.close()
# 在控制台输入以下命令后Ctrl+点击链接进入浏览器查看 tensorboard --logdir # 改变端口 tensorboard --logdir= logs --port=6007
运行时必须在生成的log文件的上层目录运行,否则不能打开
No dashboards are active for the current data set.
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)