Pytorch学习（2）TensorBoard使用：writer.add_image()函数

Wingsang

已于 2022-10-22 21:16:09 修改

阅读量1.5k

点赞数 2

分类专栏： Pytorch学习文章标签： pytorch 学习 python

于 2022-10-22 21:14:42 首次发布

原文链接：https://www.bilibili.com/video/BV1hE411t7RN?p=9&vd_source=088788d4eea5249555cfbd5040e9fbca

版权

Pytorch学习专栏收录该内容

6 篇文章 0 订阅

订阅专栏

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

writer = SummaryWriter("logs")
img_path = "dataset/train/ants_image/0013035.jpg"
img_PIL = Image.open(img_path)
img_array= np.array(img_PIL)   #将图片数据转换成numpy型才能被tensorboard读取

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

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

#writer.add_scalar()

writer.close()