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”这个位置填写日志文件

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

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐