栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

学习笔记4:pytorch的Tensorboard的使用

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

学习笔记4:pytorch的Tensorboard的使用

前言

TensorBoard的用处就是让我们更好的观测到数据,对比各阶段实验数据的变化。

一、头文件
from torch.utils.tensorboard import  SummaryWriter
import cv2
from PIL import Image
import numpy as np
二、画一个简单的y=3x的函数图像,然后查看它

1、代码:

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

其中:
writer=SummaryWriter("logs")中的logs只是文件名,是什么都可以。

2、怎么看呢?
①点击运行
②在终端输入:tensorboard --logdir=logs
③点击运行结果出现的链接

④可以看到y=3x的函数图像

三、加入图片

1、图片的格式:
①加入的图片得是tensor或者numpy型的变量

②默认图片的shape为(3,H,W)(3通道,高度,宽度)。若图片的shape为(H,W,3)也可以,但需以如下方式修改。

注:可以用print(img.shape)来查看图片的shape
2、代码:

writer=SummaryWriter("logs")
img_path='dataset/train/ants_image/0013035.jpg'
img_PIL=Image.open(img_path)
img_array=np.array(img_PIL)
writer.add_image("test",img_array,1,dataformats='HWC')

其中:
writer.add_image("test",img_array,1,dataformats='HWC')中的1是step的意思,代表test图像集合里的第一张图像,想再加图改数字就行。

运行后打开TensorBoard即可看到新增的图像

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/739119.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号