tensorboard的使用(排坑)

1.安装tensoboard

在激活对应环境的conda里面输入:(需要提前安装好pytorch,安装方法:https://blog.csdn.net/m0_53971297/article/details/139231657?spm=1001.2014.3001.5501)

pip install tensorboard

2.绘制图像

创建一共y=x的图像:

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

3.通过tensorboard查看图片

再tensorboard里面查看图片:

  1. from torch.utils.tensorboard import SummaryWriter  
    import numpy as np  
    from PIL import Image  
      
    writer =SummaryWriter("logs")  
    image_path =""#放图片的路径  
    img_PIL =Image.open(image_path)  
    img_array = np.array(img_PIL)  
    print(img_array.shape)  
    writer.add_image("test",img_array,1)  
      
    #writer.add_scalar( )  
    #y=x  
    for i in range(100):  
        writer.add_scalar("y=2x",2*i,i)  
      
    writer.close()  

如果image_path 报错:

1,更改为绝对路径;

2.在一个新的地址里面打开这个dataset;

3.用/不用\。

writer.add_image("test",img_array,1,dataformats='HWC'更改“”里面的名字,变成不同命名种类;更改里面的数字,可添加多个图片到不同step通过print(img_array.shape)查看图片形式 如果不算3HW就需要添加dataformats形式

4.打开tensorboard

运行上面的代码后,在terminal 里面输入:

tensorboard --logdir=“logs的绝对路径“

下面弹出tensorboard链接,直接=logs可能打开没有图像,变成“logs绝对路径,不能有中文“

更改端口方法:tensorboard --logdir=“logs的绝对路径“ –port=6007

如果出现图像混乱,可以删去logs文件下所有文件重新跑

  • 5
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值