Tensorboard一开始只在TensorFlow中可以使用，但是在pytorch1.1之后，pytorch中也可以进行使用。

一、安装Tensorboard

1.点击pycharm 下方的 Terminal（终端）

2.点击Terminal 右边的 “+” 选择 “command prompt ” 打开虚拟环境位置

3.输入“pip install tensorboard”，按下回车键，即可开始安装

二、Tensorboard的常用类

• SummaryWriter

1. SummaryWriter类使用方式

SummaryWriter中最主要的两个函数分别是：

writer.add_image() #写入图片

writer.add_scalar() #写入图表

下面将分类进行解释

1）writer.add_scalar() 写入图表

（1）建立图表y=x

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")#对应时间文件存储在logs文件夹之下

for i in range(100):
    writer.add_scalar(tag="y=x",scalar_value=i,global_step=i) #写入图表
    '''
        tag,图标标题
        scalar_value, y轴
        global_step=None, x轴
    '''

writer.close()

（2）点击运行，左侧弹出logs事件文件，表示运行结束

（3）打开logs文件夹下文件。在虚拟环境终端中输入如下指令：

tensorboard --logdir=<事件文件夹名称>

(4) 点击上图中的蓝色网址，即可看到y=x图表

2）writer.add_image() 写入图片

（1）编写程序导入图片

目标实现结果：

train窗口：step=1，呈现一张图片

val窗口：step=2，呈现两张图片

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


writer = SummaryWriter("logs")#对应时间文件存储在logs文件夹之下

image_path="dataset/val/ants/800px-Meat_eater_ant_qeen_excavating_hole.jpg" #任意图片位置
img_PIL=Image.open(image_path)#写入图片
img_array=np.array(img_PIL)#图片类型转换

image_path2="dataset/val/ants/8124241_36b290d372.jpg"#任意图片位置
img_PIL2=Image.open(image_path2)
img_array2=np.array(img_PIL2)

image_path3="dataset/train/bees/16838648_415acd9e3f.jpg"#任意图片位置
img_PIL3=Image.open(image_path3)
img_array3=np.array(img_PIL3)


writer.add_image("val",img_array,1,dataformats='HWC') #写入图片，numpy图片要求格式为，HWC
writer.add_image("val",img_array2,2,dataformats='HWC') #写入图片
writer.add_image("train",img_array3,1,dataformats='HWC') #写入图片

'''
            tag (str): Data identifier 图像名称，如希望单独显示，设置多个名称即可
            img_tensor (torch.Tensor, numpy.ndarray, or string/blobname)
            : 图像数据类型。注意这里需要tensor或者numpy类型的图像
            global_step (int): Global step value to record 第几步，
            walltime (float): Optional override default walltime (time.time())
              seconds after epoch of event
            dataformats (str): Image data format specification of the form
              CHW, HWC, HW, WH, etc.
'''
writer.close()

（2）点击运行，左侧弹出logs事件文件，表示运行结束

（3）打开logs文件夹下文件。在虚拟环境终端中输入如下指令：

tensorboard --logdir=<事件文件夹名称>

(4) 点击上图中的蓝色网址，即可看到图片显示结果

感谢：PyTorch深度学习快速入门教程（绝对通俗易懂！）【小土堆】

视频网址：https://www.bilibili.com/video/BV1hE411t7RN?p=9&vd_source=5b6e0605c1ed0f1db9c92503dd5994e0