深度学习pytorch之tensorboard和transform的使用

这样操作是引入tensorboard,申明一个类,logs是生成日志的文件夹,事件就在这里产生。

writer为申明的实例,这里做的画线操作 第一个是tags是图片的标签,第二个参数是y值,第三个是步长,x值

关闭writer编写

这里是将tensorboard里面加入图片的显示,HWC指定图片类型是长宽通道数,不然只能加入tensor型后面会讲 如何将tensorform来转换图片类型

在同一个tag标签下面 两个图

transform的功能

就转换图片类型大小等的一个工具

python 复制代码
from torchvision import transforms
import  cv2 as cv
from  torch.utils.tensorboard import  SummaryWriter

img_path = "dataset2/train/ants_image/0013035.jpg"

img = cv.imread(img_path)

writer = SummaryWriter("logs")


#使用tensor来将图片转成tensor类型
tensor_trans = transforms.ToTensor()#创建一个class 用作成工具 转换成tensor型
tensor_img = tensor_trans(img)#使用类的函数 result = tool(input)

writer.add_image("tensor_img",tensor_img,1)
writer.close()
相关推荐
Sui_Network11 分钟前
备受期待的 POP 射击游戏 XOCIETY 正式在 Epic Games Store 开启体验
人工智能·游戏·rpc·区块链·量子计算·graphql
漫长的~以后24 分钟前
GPT-5.2深度拆解:多档位自适应架构如何重塑AI推理效率
人工智能·gpt·架构
爱笑的眼睛1129 分钟前
自动机器学习组件的深度解析:超越AutoML框架的底层架构
java·人工智能·python·ai
LCG米32 分钟前
嵌入式Python工业环境监测实战:MicroPython读取多传感器数据
开发语言·人工智能·python
努力的BigJiang1 小时前
Cube-slam复现及报错解决
人工智能
ComputerInBook1 小时前
代数基本概念理解——特征向量和特征值
人工智能·算法·机器学习·线性变换·特征值·特征向量
漫长的~以后1 小时前
Edge TPU LiteRT V2拆解:1GB内存设备也能流畅跑AI的底层逻辑
前端·人工智能·edge
星火10241 小时前
“重生”之我用 Solo 写了一盘中国象棋
人工智能·ai编程
祝余Eleanor1 小时前
Day37 模型可视化与推理
人工智能·python·深度学习
是Dream呀1 小时前
【openFuyao】openFuyao社区AI推理加速组件技术解析与实践
人工智能·架构·openfuyao