学习视频——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
点击蓝色链接
生成图片展示柜?