PyTorch快速入门教程【小土堆】之TensorBoard的使用
视频地址P7. TensorBoard的使用（一）_哔哩哔哩_bilibili
tensorboard的用法，在左图显示2974步时输入的图像，右图是训练过程中loss的变化
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer = SummaryWriter("logs") #创建logs文件夹，把后续文件写入此文件夹中
image_path = "data/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
writer.add_image("test", img_array, 1, dataformats='HWC') #里面参数分别为，表头，要求的图像数据类型，步数，数据格式为h高w宽c通道数需要特别声明
#绘制一个y=x的图像
for i in range(100):
writer.add_scalar("y=x", i, i,) #里面参数分别为，表头，x轴，y轴
# 查看图像需要在终端输入tensorboard --logdir=logs
writer.close()