在pytorch中TensorBoard的使用

python 复制代码
from torch.utils.tensorboard import SummaryWriter

# writer的使用
# 创建实例
writer = SummaryWriter("logs")
# 存储位置在logs
# 使用如下的两种方法
# writer.add_image()
# 标量

for i in range(100):

    writer.add_scalar("y = x", i, i)

writer.close()

在终端中,对tensorboard的显示页面中,首先输入了如下的一些代码:

python 复制代码
tensorboard --logdir=logs --port=6007

在--logdir=logs中,这里出现了问题,应该等于的是如下的地址:

然后粘贴如下:

并且需要注意的是,在这个地址中,需要将地址中的斜划线改为从右至左斜方向。然后即可点击蓝色连接,即可打开:

如果画出的图,是一个三角形似的,滑动左侧的Smoothing为0.(此处已被修改,无图展示)

修改之后即可看到所需图像:

如果在实验中,想要清空之前的记忆(之前的图会一起画在一个图中),需要将logs下的event删除清零,再次运行,并输入图中正确的查看代码即可。

python 复制代码
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
# writer的使用
# 创建实例
writer = SummaryWriter("logs")
image_path = "dataset/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)

# 存储位置在logs
# 使用如下的两种方法
writer.add_image("test", img_array, 1, dataformats='HWC')

# 标量
for i in range(100):

    writer.add_scalar("y = x", 3*i, i)

writer.close()
相关推荐
SHIPKING3938 分钟前
【云端深度学习训练与部署平台】AutoDL连接VSCode运行深度学习项目的全流程
人工智能·深度学习·autodl
Q_Q51100828532 分钟前
python的婚纱影楼管理系统
开发语言·spring boot·python·django·flask·node.js·php
JNU freshman33 分钟前
计算机视觉 之 经典模型汇总
人工智能·计算机视觉
苏苏susuus37 分钟前
NLP:RNN文本生成案例分享
人工智能·rnn·自然语言处理
若兰幽竹1 小时前
【从零开始编写数据库:基于Python语言实现数据库ToyDB的ACID特性】
数据库·python
东方佑1 小时前
仅27M参数!SamOutVX轻量级语言模型刷新认知,小身材也有大智慧
人工智能·语言模型·自然语言处理
东临碣石821 小时前
【AI论文】OmniPart:基于语义解耦与结构连贯性的部件感知三维生成
人工智能
xiaocainiao8811 小时前
Python 实战:构建 Git 自动化助手
git·python·自动化
安特尼1 小时前
Datawhale AI夏令营:基于带货视频评论的用户洞察挑战赛
机器学习·语言模型·音视频
啾啾Fun1 小时前
咨询导览,AI发展趋势
人工智能