Tensorboard的使用——02单张图片add_image

Miles强

已于 2022-05-15 13:47:10 修改

阅读量636

点赞数

分类专栏： tensorboard

于 2022-05-15 13:32:18 首次发布

本文链接：https://blog.csdn.net/weixin_48501651/article/details/124781112

版权

pytorch 深度学习 python tensorflow

tensorboard 专栏收录该内容

4 篇文章

订阅专栏

这段代码展示了如何使用PyTorch的SummaryWriter在Tensorboard中添加并显示图片日志。首先实例化SummaryWriter，然后打开并转换图片为tensor，最后通过writer.add_image方法将图片添加到日志中。在终端中使用`tensorboard --logdir=logs`启动Tensorboard查看日志。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、实例化SummaryWriter，并指明日志存放路径

writer = SummaryWriter("logs")

2、打开图片

img_path = "data/train/ants_image/20935278_9190345f6b.jpg"
img = Image.open(img_path)

3、将图片转换成tensor类型

tensor_train = transforms.ToTensor()
tensor_img = tensor_train(img)

也可以使用transforms.Compose将多个transforms变化组合在一起
4、调用实例，add图片
add_image接收numpy array, torch tensor, or caffe2 blob这几种类型，所以导入图片前因转换图片类型

writer.add_image("tensor_img",tensor_img)

5、关闭writer

writer.close()

6、在terminal中使用

tensorboard --logdir=logs

代码如下：

from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter
from PIL import Image

writer = SummaryWriter("logs2")

img_path = "data/train/ants_image/20935278_9190345f6b.jpg"
img = Image.open(img_path)

tensor_train = transforms.ToTensor()
tensor_img = tensor_train(img)

writer.add_image("tensor_img",tensor_img)
writer.close()