tensorboard使用2

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

writer = SummaryWriter("logs")
img_path = "dataset/train/ants/0013035.jpg"
img_PIL = Image.open(img_path)
img_array = np.array(img_PIL)

print(type(img_array))
# 输出(512, 768, 3)分别为高宽通道
print(img_array.shape)

# 注意，从PIL到numpy，需要在add_image中指定shape中的每一个数字/维的含义
writer.add_image("test", img_array, 1, dataformats='HWC')


writer.close()

结果

参考课程视频地址：https://www.bilibili.com/video/BV1hE411t7RN?p=9