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

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()