pytorch_数据展示_SummaryWriter的两个函数
1、writer.add_scalar(“y=3x”,3*i,i)书写坐标图from torch.utils.tensorboard import SummaryWriterwriter =SummaryWriter("logs")#writer.add_image()for i in range(100):writer.add_scalar("y=3x",3*i,i)writer.close()
·
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”这个位置填写日志文件
在命令行输入这行代码,就可以获得网址
更多推荐
已为社区贡献1条内容
所有评论(0)