TensorBoard的使用

一、TensorBoard的安装

进入anaconda prompt命令行，激活pytorch，下载TensorBoard

conda activate pytorch
pip install tensorboard

二、代码实例

1、绘制一个y=x图像

from torch.utils.tensorboard import SummaryWriter

# 实例化writer，将tensorboard文件存放到logs中
writer = SummaryWriter("logs")

# 添加标量
# 例：y = x
for i in range(100):
    writer.add_scalar("y=2x", 2*i, i)
    
writer.close()

在pycharm终端中指定启动端口

tensorboard --logdir=logs --port=6007

运行之后打开连接：http://localhost:6007/

2、上传数据图像

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

# 实例化writer，将tensorboard文件存放到logs中
writer = SummaryWriter("logs")

# 添加图片
#添加图片路径
image_path = "train/ants_image/0013035.jpg"
#创建PIL的图片类
img_PIL = Image.open(image_path)
#将数据转化为img_tensor所需要的参数类型
img_array = np.array(img_PIL)
#img_tensor的形状要求格式(3,H,W),3通道，Height,Width
writer.add_image("test", img_array, 1, dataformats='HWC')

writer.close()

在pycharm终端中指定启动端口

tensorboard --logdir=logs --port=6007

运行之后打开连接：http://localhost:6007/

本文测试数据及代码，来自B站UP：我是土堆