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()
相关推荐
七宝大爷5 分钟前
NVIDIA Blackwell Ultra GB300深度解析:AI芯片性能的新巅峰
人工智能·gpu·gb300
鲸鱼在dn5 分钟前
大语言模型的后训练与“灾难性遗忘”问题——李宏毅2025大模型第六讲笔记
人工智能·笔记·语言模型
a20063801219 分钟前
ply(python版本的flex/bison or Lex/Yacc)
python
wokaoyan198126 分钟前
逻辑推演题——谁是骗子
python
九年义务漏网鲨鱼27 分钟前
利用AI大模型重构陈旧代码库 (Refactoring Legacy Codebase with AI)
python
滑水滑成滑头39 分钟前
**标题:发散创新:智能交通系统的深度探究与实现**摘要:本文将详细
java·人工智能·python
闭着眼睛学算法1 小时前
【双机位A卷】华为OD笔试之【哈希表】双机位A-跳房子I【Py/Java/C++/C/JS/Go六种语言】【欧弟算法】全网注释最详细分类最全的华子OD真题题解
java·c语言·c++·python·算法·华为od·散列表
海云安1 小时前
海云安入选安全牛《企业级AI大模型落地实战技术应用指南(2025版)》优秀案例
人工智能·安全
周杰伦_Jay1 小时前
【PaddleOCR深度解析与DeepSeek-OCR对比】开源OCR工具库的技术路线与场景适配
人工智能·机器学习·云原生·架构·开源·ocr
无限码力1 小时前
华为OD技术面真题 - Python开发 - 2
python·华为od·华为od技术面真题·华为od技术面八股·华为od技术面python八股·华为od面试python真题·华为odpython八股