2.pytorch-TensorBoard使用

b站小土堆pytorch教程学习笔记

一、两个函数：

1.writer.add_scalar()

对应的两个参数分别是损失图的横纵坐标如下图所示。

版本不兼容问题：numpy pytorch TensorFlow tensorboard等 很复杂 适当降低TensorFlow版本即可

from torch.utils.tensorboard import SummaryWriter#ctrl+类查看类详细代码

#1.初始化类
writer=SummaryWriter('logs')

#2.假设y=x
for i in range(100):
    writer.add_scalar('y=x',i,i)
writer.close()

上述函数运行结束后，产生logs文件，接下来读取logs文件

#logdir=事件文件所在文件夹名
tensorboard --logdir=logs --port=6007

2.add_image()

from torch.utils.tensorboard import SummaryWriter#ctrl+类查看类详细代码
import numpy as np
from PIL import Image

#1.初始化类
writer=SummaryWriter('logs')

image_path='dataset_another/train/ants_image/0013035.jpg'
img_PIL=Image.open(image_path)
img_array=np.array(img_PIL)
# print(img_array.shape)
#3.将会使用两个常用方法
writer.add_image('test',img_array,2,dataformats='HWC')
#第一个参数：标题
#第二个参数：图片内容
#第三个参数：步数
#第四个参数：将图片维度转变一下将通道放到前面



#2.假设y=x
for i in range(100):
    writer.add_scalar('y=x',i,i)
# writer.add_scalar()

writer.close()

报错：AttributeError: module ‘PIL.Image’ has no attribute ‘ANTIALIAS’
将Pillow降低到9.5.0