【PyTorch教程】P8-9 Tensorboard使用

完整目录

完整笔记

  • 使用这个工具,可以看到具体某一步骤时的输入和输出。
    在这里插入图片描述

  • 需要指定一个文件夹,把创建的事件文件存下来。这是Summarywriter的init当中做的。

  • 还有其他参数可以设置,略了。

  • 一般使用这两个方法:
    在这里插入图片描述
    注意add_scalar里面的参数,代表了图中横纵坐标的轴名称:
    在这里插入图片描述
    举例子:SummaryWriter是个class,需要将它实例化:
    在这里插入图片描述

  • 如何打开浏览器看图?
    Port那里是为了避免跟别人冲突,自己定义一个数值:
    在这里插入图片描述
    如果出现“刷新之后产生了以外变化”,就把原来的logs文件夹删掉,重新运行就好了:
    在这里插入图片描述

  • add_image方法的使用:注意输入类型的要求:
    print(type(img)),PIL读取的类型不行;

  • 视频中,用PIL转成了numpy类型作为输入了。
    在这里插入图片描述
    在这里插入图片描述

  • 以上代码出的问题是:通道(H,W,C)需要转换:
    那个20行最后一个参数 dataformats 可以把这个顺序转换过来。

可以运行的代码

# !usr/bin/env python3
# -*- coding:utf-8 -*-

"""
author :24nemo
 date  :2021年07月07日
"""

import numpy as np
from PIL import Image
from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")
image_path = "data/train/ants_image/6240329_72c01e663e.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)

writer.add_image("train", img_array, 1, dataformats='HWC')  # 数字 1 代表一共有几个步骤,网页中会出现滑块,可以拖动
# y = 2x
for i in range(100):
    writer.add_scalar("y=2x", 3 * i, i)

writer.close()
'''

终端:
tensorboard --logdir=logs --port=6007

logs文件可以删除,重新运行

'''

完整目录

  • 11
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值