tensorboard使用

# Tensorboard可以用来查看loss是否按照我们预想的变化，或者查看训练到某一步输出的图像是什么样
import os.path

import numpy as np
from PIL import Image
from torch.utils.tensorboard import SummaryWriter

# help(SummaryWriter)
# Tensorboard写日志
# 创建一个logs文件夹，writer写的文件都在该文件夹下
writer=SummaryWriter("logs")
for i in range(100):
    writer.add_scalar("y=2x",2*i,i)
writer.close()

# Tensorboard读日志
# tensorboard --logdir=路径地址 --port=端口名
# 运行上面命令会获得一个网址，输入网址可得Tensorboard界面

# Tensorboard读图片
img_path1 = "D:/05深度学习笔记/Data/FirstTypeData/train/ants/0013035.jpg"
img_PIL1 = Image.open(img_path1)
img_array1 = np.array(img_PIL1)
img_path2 = "D:/05深度学习笔记/Data/FirstTypeData/train/bees/17209602_fe5a5a746f.jpg"
img_PIL2 = Image.open(img_path2)
img_array2 = np.array(img_PIL2)

writer = SummaryWriter("logs")
writer.add_image("test", img_array1, 1, dataformats="HWC")
writer.add_image("test", img_array2, 2, dataformats="HWC")
writer.close()