TensorBoard的使用

TensorBoard的使用

打开项目文件

第一步，打开项目文件，新建py文件，设置环境变量，导包

from torch.utils.tensorboard import SummaryWriter

安装tensorboard

方法一：在cmd终端安装

pip install tensorboard

方法二：在pycharm控制台中安装

生成logs文件

简单举例y=x的直线绘制，书写函数之后可以按住CTRL键查看该函数如何定义的，自己设置传入的参数

from torch.utils.tensorboard import SummaryWriter
writer =SummaryWriter("logs")
# writer.add_image()
for i in range(100):
    writer.add_scalar("y=x",i,i)
writer.close()

运行上段代码之后会在根目录中生成一个logs文件夹

在控制台中输入命令打开文件夹
tensorboard --logdir=事件文件所在的文件夹名
或者可以指定端口名
tensorboard --logdir=事件文件所在的文件夹名 --port=xxxx

add_image的使用（常用来观察训练结果）

在项目文件夹中放入图片，可以是自己的也可以是各种数据集
1.定义数据路径

img_path="flower_photos/daisy/5547758_eea9edfd54_n.jpg"

2.读入数据路径

img_PIL=Image.open(img_path)

3.将数据类型转换为numpy型

img_array=np.array(img_PIL)

4.输出训练结果

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

如果数据的类型是HW通道，要在传参时说明dataformats=‘HWC’

最终输出