文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

【人工智能概论】 PyTorch可视化工具Tensorboard安装与简单使用

2023-10-02 17:58

关注

【人工智能概论】 PyTorch可视化工具Tensorboard安装与简单使用

文章目录


一. 安装Tensorboard

1.1 安装Tensorboard

ModuleNotFoundError: No module named ‘tensorboard’

pip install tensorboard

1.2 验证安装

from torch.utils.tensorboard import SummaryWriter

二.Tensorboard功能简介

  1. Tensorboard常用的四个功能
  1. 画线——add_scalaar
from torch.utils.tensorboard import SummaryWriter# 创建编辑器,保存日志,指令保存路径log_dirwriter = SummaryWriter(log_dir="./logs") # 指定保存位置# y = 2 * xfor i in range(100):    # 添加,x轴,y轴    # tag: 名, scalar_value: y轴, global_step: x轴    writer.add_scalar(tag="y=2x",scalar_value=2*1,global_step=i)# 关闭writer.close()
# 使用默认端口tensorboard --logdir=logs# 使用指定端口tensorboard --logdir=logs --port=6007
  1. 展示单张图像——add_image
from torch.utils.tensorboard import SummaryWriterimport numpy as npfrom PIL import Image# 创建编辑器,保存日志,指令保存路径log_dirwriter = SummaryWriter(log_dir="./logs") # 指定保存位置# 图像地址image_path = "/home/chenshili/图片/food_01.jpeg"# 打开图像img_PIL = Image.open(image_path) # 用PIL打开的图像,其文件格式是PIL的特有类# 转化成numpy数据,因为add_image只能传入array或者tensor数据,所以要做类型转化。img_array = np.array(img_PIL)# 开始画图,tag: ,img_tensor: tensor或者numpy类型的数据, dataformats: H高W宽C通道,指定HWC类型writer.add_image(tag="train",img_tensor=img_array,global_step=1,dataformats="HWC")# 关闭writer.close()
  1. 画模型——add_graph
from torch.utils.tensorboard import SummaryWriterimport torchfrom torch.nn import Linear,ReLU,Sequential# 创建编辑器,保存日志,指令保存路径log_dirwriter = SummaryWriter(log_dir="./logs") # 指定保存位置model = Sequential(    Linear(1,10),    ReLU(),    Linear(10,1))data = torch.ones(10,1)writer.add_graph(model=model,input_to_model=data)# 关闭writer.close()
  1. 模型对比
from torch.utils.tensorboard import SummaryWriterimport osimport datetime# 按时间构建生成文件保存的文件夹log_dir = os.path.join("./logs",datetime.datetime.now().strftime("%Y%m%d_%H%M%S"))print(log_dir)# 创建编辑器,保存日志,指令保存路径log_dirwriter = SummaryWriter(log_dir=log_dir) # 指定保存位置for i in range(100):    writer.add_scalar(tag="loss",scalar_value=1*i,global_step=i)    # 多执行几次把scalar_value的值做修改,如2*i,i等# 关闭writer.close()

来源地址:https://blog.csdn.net/qq_44928822/article/details/128722693

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯