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.
