pytorch_数据展示_SummaryWriter的两个函数

1、导入库函数

from torch.utils.tensorboard import SummaryWriter

2、首先实例化Writer

writer = SummaryWriter(log_dir='logs',flush_secs=60)

• log_dir：tensorboard文件的存放路径
• flush_secs：表示写入tensorboard文件的时间间隔

3、writer.add_scalar(“y=3x”,3*i,i)书写坐标图

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

#接下来的操作在pycharm的终端进行。
 tensorboard --logdir=logs --port=6007#在这个位置port设置终端的端口号

上面的主要用到的是writer.add_scalar(标题，x的变化，y的变化)

4、writer.add_image()书写

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer =SummaryWriter("logs")
image_path="练手数据集/train/bees_image/16838648_415acd9e3f.jpg"
img_PIL = Image.open(image_path)
img_array=np.array(img_PIL)
writer.add_image("test",img_array,2,dataformats='HWC')

参数：
tag (string): Data identifier 标题
img_tensor : Image data 在这个位置输入图像且图像的数据类型必须是tensor、numpy等类型的
global_step (int): Global step value to record 训练的步骤

5、tensorboard --logdir=“logs”这个位置填写日志文件

在命令行输入这行代码，就可以获得网址