Pytorch学习日记——Tensorboard使用（二）

学习视频——B站【小土堆】

代码

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

writer = SummaryWriter("logs")
image_path = "dataset/train/erciyuan_image/871-28510.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)

writer.add_image("test", img_array, 3, dataformats='HWC')
#y = 2x
for i in range(100):

    writer.add_scalar("y=2x",2*i,i)

writer.close()

image_path = 为图片的相对地址

riter.add_image("test", img_array, 3, dataformats='HWC')

 此代码中，3为加载图片的步骤，dataformats='HWC'为如果你的图片不是默认规格，需要写上这段话，具体的可以参考add_image这个函数。

点击运行，跑通后在Terminal输入

tensorboard --logdir=logs --port=6007

点击蓝色链接

生成图片展示柜？