DL - TensorBoard

TensorBoard

---

一、安装 TensorBoard

(base) $ sudo pip install tensorboard==1.15.0

tensorflow 1.14.0 requires tensorboard<1.15.0,>=1.14.0, but you have tensorboard 1.15.0 which is incompatible.

ImportError: TensorBoard logging requires TensorBoard version 1.15 or above

---

启动

启动，并修改 tensorboard 端口

需要进入 logs 所在根目录

tensorboard --logdir=logs --port=6007

访问：http://127.0.0.1:6007

---

二、使用

1、显示坐标轴

import os
from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter('logs')

# writer.add_image()
for i in range(100):
    # lable, y, x
    writer.add_scalar('y=2x', i * 2, i)


writer.close()

---

2、展示图片

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

writer = SummaryWriter('logs')

img_path = '/Users/user/Pictures/美图/001.jpeg'
img_pil = Image.open(img_path)
img_arr = np.array(img_pil) # numpy 型数据

print(type(img_arr), img_arr.shape) # <class 'numpy.ndarray'> (799, 1198, 3)

writer.add_image('test2', img_arr, 1, dataformats='HWC')
 
writer.close()

---

---

可以修改图片，统一标签下添加另一张图片，step 为2；

在 tensorboard 上可以滑动显示图片。

---

使用注意

如果同一个标签下，使用了两个函数；

会出现两个函数的结果，并产生拟合

需要将对应的 log 文件删除，然后重启 tensorboard 程序

官方建议重新写一个子文件。