PyTorch教程【五】TensoBoard的使用

一、安装TensorBoard

1、进入Anaconda Prompt，激活环境conda activate pytorch(或直接在PyCharm中打开Terminal终端)

2、输入命令pip install tensorboard

3、安装成功

二、代码示例

  from torch.utils.tensorboard import SummaryWriter

  writer = SummaryWriter("logs")

  # writer.add_image()
  # y = x
  for i in range(100):
      writer.add_scalar("y=x", i, i)

  writer.close()

三、打开事件文件

  tensorboard --logdir=logs --port=6007

logdir=事件文件所在的文件夹名
port=指定端口

打开对应网址

四、利用Opencv读取图片，获得numpy型图片数据

（1）安装Opencv，通过命令pip install opencv-python，如果安装缓慢，可以更换pip源，更换方法参考pip 换源

（2）代码

  import cv2
  cv_img=cv2.imread(img_path)

五、利用numpy.array(),对PIL图片进行转换

  import numpy as np
  img=np.array(img)

从PIL到numpy，需要在add_image()中指定shape中每一个数字/维表示的含义

六、add_image()的使用（常用来观察训练结果）

  from torch.utils.tensorboard import SummaryWriter
  import numpy as np
  from PIL import Image
  
  writer = SummaryWriter("logs")
  img_path = "data/train/ants_image/0013035.jpg"
  img_PIL = Image.open(img_path)
  img_array = np.array(img_PIL)

  writer.add_image("test", img_array, 1, dataformats='HWC')