作用：用于展示某个值或者图像的变换过程

用于展示值的变化

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project ：Pytorch学习 
@File    ：tensorboard.py
@IDE     ：PyCharm 
@Author  ：咋
@Date    ：2023/6/29 15:59 
"""
'''
from torch.utils.tensorboard.writer import SummaryWriter 报错
解决方法:
pip install  -i https://mirrors.aliyun.com/pypi/simple/ tensorboardX
from tensorboardX import SummaryWriter
'''

from tensorboardX import SummaryWriter
# from torch.utils.tensorboard.writer import SummaryWriter

writer = SummaryWriter("logs")
# writer.add_image()
for i in range(100):
    writer.add_scalar("y=2x",2*i,i)
writer.close()

'''
开启指令：
tensorboard --logdir=logs --port=6007 # port指定端口号，防止冲突  # 使用anaconda powershell 打开
'''

image.png

用于展示图像的变化

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project ：Pytorch学习 
@File    ：tensorboard_2.py
@IDE     ：PyCharm 
@Author  ：咋
@Date    ：2023/6/29 16:48 
"""
from tensorboardX import SummaryWriter
import cv2

write = SummaryWriter("log")
image = cv2.imread("data\\train\\ants_image\\5650366_e22b7e1065.jpg")
# write.add_image("test",image,1,dataformats='HWC')
write.add_image("test",image,2,dataformats='HWC')
write.close()

image.pngimage.png
上面有一个滑动条，可以拖动滑动条展示图片的变化过程，这个方法可以用在pytorch模型训练时，可视化出特征图的变化。
ps：奇怪的是用pycharm的终端和本地的终端tensorboard –logdir=logs –port=6007都进不去，anaconda powershell 能进去，而且本地的终端，社区版还有jupyter notebook的终端一运行就会跳转到专业版的pycharm，就很奇怪。没有报错，但是没有给连接地址，之后用tensorboard的话就用anaconda自带的终端进入。