tensorboard无法显示新加入的图片的原因

楚江_wog1st

已于 2022-11-23 21:49:05 修改

阅读量1k

点赞数 2

文章标签：深度学习 pytorch numpy

于 2022-09-08 19:34:02 首次发布

本文链接：https://blog.csdn.net/weixin_51940803/article/details/126770651

版权

在学习用tensorboard的时候，发现使用writers.add_image("tensor_image",tensor_image)能正常运行但是会报错，仔细思考了下得出原因：

writers在使用之后需要关闭，不然tensorboard会认为你仍在进行输入操作，从而不会给你显示，当你调用writers.close()之后，就能成功显示了。

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

img_path = "YOUR LOGS PATH"
img = Image.open(img_path)
writer = SummaryWriter("logs")

# 调用ToTensor类，创建具体工具transforms.ToTensor()
tensor_trains = transforms.ToTensor()
tensor_img = tensor_trains(img)
writer.add_image("tensor_img", tensor_img)

# 每次用完writer，都要记得把writer给关了，要不然tensorboard会认为还在输入，就不会往里头加东西
writer.close()

感谢GJmachine_code老哥的补充，除了上述问题外，还需要进行以下操作：

1 关闭tensorboard的所有网页
2 将之前的log文件给del
3 将pycharm的终端（就是控制台左边那个）给全部关闭
4 然后用以往的方法重新运行打开tensorboard就行了