Pytorch学习——Tensorboard

最新推荐文章于 2024-11-29 20:24:19 发布

原创最新推荐文章于 2024-11-29 20:24:19 发布 · 303 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据可视化 #深度学习

深度学习专栏收录该内容

6 篇文章

订阅专栏

本文介绍TensorBoard的基本使用方法，包括如何使用add_scalar函数绘制线性图表，以及如何使用add_image函数展示图片数据。通过具体代码示例，展示了如何在TensorBoard中记录和可视化训练过程中的数据。

Tensorboard是可视化工具。

1.add_scalar

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

# writer.add_image()
for i in range(100):
    writer.add_scalar("y=2x", i, 2*i)

writer.close()

找到生成文件所在位置：

点开网址：

2.add_image

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

writer = SummaryWriter("logs")
image_path = "data/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)  #转换类型

writer.add_image("img", img_array, 1, dataformats='HWC' )

writer.close()