【pytorch】（二）Tensorboard

from torch.utils.tensorboard import SummaryWriter

#初始化实例
writer = SummaryWriter("logs")

#会用到两个方法
writer.add_image()
writer.add_scalar()

writer.close()

简单示例

logdir = 事件文件所在文件夹名

for i in range(100):
	writer.add_scalar("y=x",i,i)# y = x是title，第二个参数是x，第三个参数是y
	writer.add_scalar("y=2x",2 * i,i)
witer.close()

查看log文件

#终端输入 tensorboard --logdir=logs
#logs是一个文件夹，里面有envents...文件
#点击代码中的网址http://localhost:6006/

注意：title名字一样时候，一张图可能显示多个你运行的结果，这个时候选择把event…删掉重新运行就行

add_image()方法

#---------------------小测试----------------------------------
image_path = "data/train/ants_image/001.jpg"
from PIL import Image
img = Image.open(image_path)
print(type(img))

import numpy as np
img_array = np.array(img)
print(type(img_array))

#-----------------------------正式代码-----------------------------
image_path = "data/train/ants_image/001.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)

writer.add_iamge("test", img_array, 1, dataformats='HWC')

writer.close()