【PyTorch

【PyTorch

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.