PyTorch深度学习快速入门教程【小土堆】（2）

---

---

前言

p7-p8 TensorBoard的使用
B站视频链接：PyTorch深度学习快速入门教程（绝对通俗易懂！）【小土堆】

---

一、SummaryWriter展示函数图像

代码如下

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

# Test 01
writer = SummaryWriter('logs')
for i in range(100):
    writer.add_scalar('y=2x', 2 * i, i)  # 标题  y取值  x取值

writer.close()

点击pycharm下方的Terminal输入 命令tensorboard --logdir=logs，之后点击链接即可

二、SummaryWriter展示图片

代码如下

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

# Test 02
writer = SummaryWriter('logs')
image_path = 'dataset/train/ants/0013035.jpg'  # 相对路径
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(img_array.shape)
writer.add_image("test", img_array, 1, dataformats='HWC')  # 标题 图片 步骤  通道顺序
writer.close()

刷新后效果如下（如果刷新没有效果可以先在Terminal按住Ctrl+C关闭后再次执行命令）