TensorBoard

TensorBoard的含义

显示训练时的曲线图与每一步的图像

TensorBoard的使用

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

image_path = "data/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)#将PIL转化为np.array
writer = SummaryWriter("logs")#存放信息的路径

writer.add_image("test", img_array, 1, dataformats='HWC')#显示图片
for i in range(100):

    writer.add_scalar("y=2x", i, i)#显示y=2x的曲线图

writer.close()

## 参数解析
writer.add_scalar（“图表标题”，数值(y)，步数(x)）
writer.add_image（“图表标题”，数值(y)，步数(x),dataformats='HWC'）
将图片转化为（高度(H)，宽度(W)，尺寸(C)）的格式
## 图表显示

命令行输入：tensorboard --logdir=logs
指定端口：tensorboard --logdir=logs --port=端口号