TensorBoard展示数据

TensorBoard是一个可视化工具,用于帮助机器学习工程师和研究人员可视化训练过程中的模型、数据和指标。

一、add_scalar:

from torch.utils.tensorboard import SummaryWriter

#文件存储到logs文件夹下
writer = SummaryWriter("logs")
#添加标量数据到Summary
#tag
#def add_scalar(self,tag,scalar_value,global_step=None,walltime=None,new_style=False,double_precision=False)
for x in range(10):
    writer.add_scalar("y=2x",x,2*x)

writer.close()

二、add_image:

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

#文件存储到logs文件夹下
writer = SummaryWriter("logs")

image_path = "dataset/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_ndarray = np.array(img_PIL)

writer.add_image("img", img_ndarray,1,dataformats="HWC")
writer.close()

三、查看文件内容:

在这里插入图片描述

相关推荐

  1. pytorch通过 tensorboardX 调用 Tensorboard 进行可视化

    2024-07-16 19:44:03       25 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-16 19:44:03       67 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-16 19:44:03       71 阅读
  3. 在Django里面运行非项目文件

    2024-07-16 19:44:03       58 阅读
  4. Python语言-面向对象

    2024-07-16 19:44:03       69 阅读

热门阅读

  1. JVM参数调优经验

    2024-07-16 19:44:03       21 阅读
  2. conda配置虚拟环境的常用命令

    2024-07-16 19:44:03       17 阅读
  3. MATLAB实现一个车辆悬架PID模拟系统

    2024-07-16 19:44:03       19 阅读
  4. python基础语法

    2024-07-16 19:44:03       19 阅读
  5. vue 项目代码架构

    2024-07-16 19:44:03       17 阅读
  6. 通过 Nginx 修复 CORS 漏洞

    2024-07-16 19:44:03       21 阅读
  7. [C++ 入门基础 - 引用]

    2024-07-16 19:44:03       20 阅读
  8. Elasticsearch:将Logstash日志存到elasticsearch中

    2024-07-16 19:44:03       21 阅读
  9. 题解:P10678 『STA - R6』月

    2024-07-16 19:44:03       16 阅读
  10. 每天一个数据分析题(四百三十)- 假设检验

    2024-07-16 19:44:03       18 阅读
  11. 【PG】PostgreSQL高可用之repmgr命令手册

    2024-07-16 19:44:03       20 阅读
  12. GPT带我学-设计模式14-工厂模式

    2024-07-16 19:44:03       18 阅读
  13. InnoDB

    2024-07-16 19:44:03       21 阅读