前言
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关闭后再次执行命令)