pytorch深度学习2: Tensorboard的使用

源代码：

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

writer = SummaryWriter("logs")
image_path = "data/train/ants_image/6240329_72c01e663e.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)

writer.add_image("train", img_array, 1, dataformats='HWC')  # 此处注意dataformats需要和图片的对应
# y = 2x
for i in range(100):
    writer.add_scalar("y=2x", 3*i, i)

writer.close()

流程：

创建 SummaryWriter 类，传入 logs 文件输出地址
图片文件地址
Image 类打开图片文件
将 Image 类转换 numpy 格式
SummaryWriter 类加载 numpy 格式图片
画图
关闭 SummaryWriter 流

运行方式：

打开 anaconda prompt，输入

conda activate pytorch

进入 pytorch 环境下 ，输入

tensorboard --logdir "文件夹绝对地址" --port=6007(optional)

然后进入 http://localhost:6006/ 查看图片和曲线