pytorch -- tensorboard使用

1. 导入

from torch.utils.tensorboard import SummaryWriter
2. ctrl键点击查看如何使用
3. writer = SummaryWriter("dir") 对应事件文件输出到文件夹dir中
4. def add_image(
self,
tag,
img_tensor,
global_step=None, 步骤
walltime=None, dataformats="CHW"
):

5. writer.add_scalar("y=x",i,i) 添加标量数据到summary中,tag:图表标题
def add_scalar(
self,
tag, 图表的标题
scalar_value, y轴数值
global_step=None, x轴数值
walltime=None,
new_style=False,
double_precision=False,
):

6. 打开事件文件 终端使用: tensorboard --logdir=logs

使用:

python 复制代码
from torch.utils.tensorboard import SummaryWriter
import cv2
writer = SummaryWriter("logs")


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

img_path = "dataset/train/bees/16838648_415acd9e3f.jpg"
img_cv = cv2.imread(img_path)
print(img_cv.shape)
writer.add_image("ants",img_cv,1, dataformats='HWC')

writer.close()
相关推荐
独行soc6 小时前
2025年渗透测试面试题总结-66(题目+回答)
java·网络·python·安全·web安全·adb·渗透测试
先做个垃圾出来………6 小时前
残差连接的概念与作用
人工智能·算法·机器学习·语言模型·自然语言处理
AI小书房7 小时前
【人工智能通识专栏】第十三讲:图像处理
人工智能
fanstuck7 小时前
基于大模型的个性化推荐系统实现探索与应用
大数据·人工智能·语言模型·数据挖掘
多看书少吃饭8 小时前
基于 OpenCV 的眼球识别算法以及青光眼算法识别
人工智能·opencv·计算机视觉
Y学院8 小时前
Python 数据分析:从新手到高手的“摸鱼”指南
python·数据分析
一条数据库8 小时前
南京方言数据集|300小时高质量自然对话音频|专业录音棚采集|方言语音识别模型训练|情感计算研究|方言保护文化遗产数字化|语音情感识别|方言对话系统开发
人工智能·音视频·语音识别
Yingjun Mo8 小时前
1. 统计推断-基于神经网络与Langevin扩散的自适应潜变量建模与优化
人工智能·神经网络·算法·机器学习·概率论