pytorch框架学习(tensorboard的使用)

什么是tensorboard？

tensorboard是一个可视化工具，它可以把训练过程中的数据变化以图像的形式绘制出来，或者记录训练过程中使用的图片

tensorboard的安装：

在pycharm的终端中输出安装命令后自动安装——

pip install tensorboard SummaryWriter： 理解：

summarwriter是pytorch库里的一个类，用来在给定目录中创建事件文件，并向其中添加摘要和事件，以供 TensorBoard 使用

导入： from torch.utils.tensorboard import SummaryWriter add_scalar()方法：

它用于将一个标量值添加到一个记录器或日志文件中

主要参数：标签，scalar_value，global_step

标签就是图的说明（表头），scalar_value简单理解为y轴值（函数值），global_step简单理解为x轴值（变量值）

绘制并查看：

代码：

from torch.utils.tensorboard import SummaryWriter#导入SummaryWriter类 writer = SummaryWriter("logs") # 用SummaryWriter示例化writer对象，用于记录日志 # 假设绘制y = 2*x的图像 for i in range(100): # 参数：标签 , 函数值 , 变量值 writer.add_scalar("y=2*x", 2 * i, i) #注意：y=2*x的含义为函数值是变量值的两倍，故参数二是2*i，参数三是i writer.close()#回收对象

运行结果： 

接着需要打开这个日志文件，步骤如下：

①在终端输入如下命令:

tensorboard --logdir=logs

②得到网址，默认端口为6006，点击直接跳转，得到绘制好的图像

③（可选）输入以下命令可以指定生成结果的端口：

tensorboard --logdir=logs --port=6007

add_image()方法：

它用于将一张图片添加到一个记录器或日志文件中

主要参数：标签，img_tensor，global_step，参数格式=' '

标签就是图片的说明，img_tensor要求是一个tensor类型或者numpy类型，因此使用PIL的Image.open()方法打开图片后还需要进行一步类型转换，如下：

import numpy as np # 导入numpy库的np工具包，为np.array()依赖 image_path = "MYDATA/train/ants_image/0013035.jpg" # 图片的相对路径 img_PIL = Image.open(image_path) # 打开图片 img_array = np.array(img_PIL) # 更改为numpy类型

再结合前面的知识点，就能对图片进行记录了，完整代码如下：

from torch.utils.tensorboard import SummaryWriter from PIL import Image import numpy as np writer = SummaryWriter("logs") # 用SummaryWriter示例化writer对象，用于记录日志 image_path = "MYDATA/train/ants_image/0013035.jpg" img_PIL = Image.open(image_path) # 打开图片 print(type(img_PIL)) # 这里是测试（打印）Image.open()返回的类型，结果为<class 'PIL.JpegImagePlugin.JpegImageFile'-> # <-这个类型不被add_image()支持，因此使用更np.array()改为numpy类型 img_array = np.array(img_PIL) print(type(img_array)) #查看转换后的类型，结果是<class 'numpy.ndarray'> print(img_array.shape) #查看图片格式，结果为(512, 768, 3)，即HWC writer.add_image("no.1", img_array, 2, dataformats = 'HWC') #注：add_image()默认图片为CHW格式，可通过dataformats = 'HWC'指定格式 writer.close()