SummaryWriter和tensorboard使用示例

最新推荐文章于 2025-05-23 19:48:30 发布

原创最新推荐文章于 2025-05-23 19:48:30 发布 · 2.8k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #开发语言 #后端

收录于

本文介绍如何利用PyTorch的TensorBoard工具记录并展示模型训练的过程，包括图像数据的加载与显示、训练指标的记录等。通过示例代码演示了如何将图像数据和训练损失等信息写入TensorBoard，并通过命令行启动TensorBoard查看记录的数据。

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

writer = SummaryWriter("logs")
image_path = "dataSet/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()

运行后产生logs文件

然后命令行输入 tensorboard --logdir 'NeuralNetwork/test/logs'