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

PyTorch入门(二)tensorboard和Transforms的使用

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

PyTorch入门(二)tensorboard和Transforms的使用

使用tensorboard展示数值或图像

绘制图像:

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs") # 存储对应的事件文件

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

执行完了之后,会出现一个logs文件夹,里面有对应的文件
需要在终端执行:

tensorboard --logdir=logs


浏览器输入这个地址,即可访问。
可以指定端口,防止端口冲突

tensorboard --logdir=logs --port=6007

显示图片:

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer = SummaryWriter("logs") # 存储对应的事件文件

# 需要先把图像转成numpy.array类型
image_path = "dataset/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
# 接收的图像只能是这几种类型 : torch.Tensor, numpy.array, or string/blobname
# 后面加参数是因为 需要指定shape格式 具体可以点进这个函数去看 写的有注释
writer.add_image("test",img_array,1,dataformats='HWC')

writer.close()

执行代码之后,还去哪个地址localhost:6006 查看图片
我们可以把图像地址换了,查看别的图像,但是注意,这样操作之后,有时候会导致,显示的图像有问题,或者不是我们期望的那一张。
解决办法有两个:
1、把logs文件夹里面的删除了,再进行别的图片的显示
2、显示别的图片的时候,新建个文件夹
3、终端重新执行一下tensorboard --logdir=logs

Transforms

Transforms主要是对图像做一些变换
比较常用的是把图像转换成tensor类型

from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms

# 读入一个img
img_path = "dataset/ants_image/0013035.jpg"
img = Image.open(img_path)

# writer = SummaryWriter("logs")

# 将其转换为tensor类型
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
print(tensor_img)

# writer.add_image("Tensor_img",tensor_img)
# writer.close()

结果如图:

还有一些其他的常用方法:
Resize:https://blog.csdn.net/qq_40714949/article/details/115393592

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

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

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