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()
相关推荐
星期天要睡觉2 分钟前
计算机视觉(opencv)——基于 MediaPipe 的手势识别系统
人工智能·opencv·计算机视觉
三年呀6 分钟前
指纹技术深度剖析:从原理到实践的全方位探索
图像处理·人工智能·计算机视觉·指纹识别·生物识别技术·安全算法
学习的周周啊41 分钟前
一人AI自动化开发体系(Cursor 驱动):从需求到上线的全流程闭环与实战清单
运维·人工智能·自动化·ai编程·全栈·devops·cursor
后端小肥肠1 小时前
明星漫画总画不像?用 Coze +即梦 4 工作流,素描风漫画3分钟搞定,小白也能上手
人工智能·aigc·coze
flay1 小时前
5个Claude实战项目从0到1:自动化、客服机器人、代码审查
人工智能
flay1 小时前
Claude API完全指南:从入门到实战
人工智能
用户5191495848451 小时前
OAuth/OpenID Connect安全测试全指南
人工智能·aigc
初级炼丹师(爱说实话版)1 小时前
PGLRNet论文笔记
人工智能·深度学习·计算机视觉
川石课堂软件测试2 小时前
全链路Controller压测负载均衡
android·运维·开发语言·python·mysql·adb·负载均衡
明月照山海-2 小时前
机器学习周报十七
人工智能·机器学习