【Pytorch 03】TensorBoard

目录

        1. 安装 TensorBoard，执行下面语句：

2. Pycharm 中的terminal无法进入pytorch环境：

3. Pycharm 中写入以下语句：

4. Pycharm 中的 terminal 中打开tensorboard

【重点】5. 在tensorboard中画出y=x图像，在tensorboard中添加图片

---

1. 安装 TensorBoard，执行下面语句：

conda activate pytorch
pip install tensorboard

2. Pycharm 中的terminal无法进入pytorch环境：

将 powershell.exe 更改成 cmd.exe

3. Pycharm 中写入以下语句：

# 作者：要努力，努力，再努力
# 开发时间：2022/5/2 10:59

from torch.utils.tensorboard import SummaryWriter

write = SummaryWriter('logs')

for i in range(100):
    write.add_scalar('y=x', i, i)

write.close()

4. Pycharm 中的 terminal 中打开tensorboard

cd tensorboard  # 必须切换到logs文件夹的上一级目录
tensorboard --logdir=logs --port=6007  # 切换端口

【重点】5. 在tensorboard中画出y=x图像，在tensorboard中添加图片

目录结构：

 

 

# 作者：要努力，努力，再努力
# 开发时间：2022/5/2 10:59

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

img_path = "../dataset/train/bees_images/29494643_e3410f0d37.jpg"  # 图片路径地址
img_PIL = Image.open(img_path)  # 打开图片地址 PIL 格式
img_array = np.array(img_PIL)  # 将 PIL 格式 的图片转成 numpy 方式
# print(img_array.shape)

write = SummaryWriter('logs')  # tensorboard写logs
write.add_image('test2', img_array, 1, dataformats='HWC')
# 看一下add_image函数使用，add_image(self, tag, img_tensor, global_step=None, walltime=None, dataformats='CHW'):

for i in range(100):
    write.add_scalar('y=x', i, i)  # 保存数据，做可视化

write.close()  # 书写关闭