动手学深度学习(pytorch土堆)-02TensorBoard的使用

1.可视化

代码使用了 torch.utils.tensorboard 将数据记录到 TensorBoard 以便可视化。具体来说,它将标量数据记录到目录 logs 中,使用的是 SummaryWriter 类。

代码分解如下:

  1. SummaryWriter("logs") :初始化一个 TensorBoard 的写入器,日志会保存到 "logs" 目录。
  2. writer.add_scalar("y=x", i, i) :在循环的每一次迭代中,写入器都会记录一个名为 "y=x" 的标量数据,标量的值和步数都等于 i。这实际上是在记录一条 y = x 的直线。
  3. writer.close():关闭写入器,确保所有数据都已写入到磁盘。

运行这段代码后,使用以下命令在 pycharm控制台中可视化结果:

bash 复制代码
tensorboard --logdir=logs

然后在浏览器中访问 localhost:6006 来查看记录的标量数据。

c 复制代码
from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter("logs")
# writer.add_image()

for i in range(100):
    writer.add_scalar("y=x",i,i)
writer.close()

2显示图像

c 复制代码
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer=SummaryWriter("logs")
image_path="hymenoptera_data/train/ants/0013035.jpg"
img_PIL=Image.open(image_path)
img_array=np.array(img_PIL)
writer.add_image("test",img_array,1,dataformats="HWC")

for i in range(100):
    writer.add_scalar("y=x",i,i)
writer.close()
相关推荐
羑悻的小杀马特2 小时前
OpenCV 引擎:驱动实时应用开发的科技狂飙
人工智能·科技·opencv·计算机视觉
guanshiyishi5 小时前
ABeam 德硕 | 中国汽车市场(2)——新能源车的崛起与中国汽车市场机遇与挑战
人工智能
极客天成ScaleFlash5 小时前
极客天成NVFile:无缓存直击存储性能天花板,重新定义AI时代并行存储新范式
人工智能·缓存
Uzuki5 小时前
AI可解释性 II | Saliency Maps-based 归因方法(Attribution)论文导读(持续更新)
深度学习·机器学习·可解释性
澳鹏Appen6 小时前
AI安全:构建负责任且可靠的系统
人工智能·安全
蹦蹦跳跳真可爱5897 小时前
Python----机器学习(KNN:使用数学方法实现KNN)
人工智能·python·机器学习
视界宝藏库7 小时前
多元 AI 配音软件,打造独特音频体验
人工智能
xinxiyinhe8 小时前
GitHub上英语学习工具的精选分类汇总
人工智能·deepseek·学习英语精选
byxdaz8 小时前
PyTorch中Linear全连接层
pytorch
Start_Present8 小时前
Pytorch 第十二回:循环神经网络——LSTM模型
pytorch·rnn·神经网络·数据分析·lstm