深度学习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()
相关推荐
埃菲尔铁塔_CV算法20 分钟前
人工智能图像算法:开启视觉新时代的钥匙
人工智能·算法
EasyCVR20 分钟前
EHOME视频平台EasyCVR视频融合平台使用OBS进行RTMP推流,WebRTC播放出现抖动、卡顿如何解决?
人工智能·算法·ffmpeg·音视频·webrtc·监控视频接入
打羽毛球吗️26 分钟前
机器学习中的两种主要思路:数据驱动与模型驱动
人工智能·机器学习
好喜欢吃红柚子43 分钟前
万字长文解读空间、通道注意力机制机制和超详细代码逐行分析(SE,CBAM,SGE,CA,ECA,TA)
人工智能·pytorch·python·计算机视觉·cnn
小馒头学python1 小时前
机器学习是什么?AIGC又是什么?机器学习与AIGC未来科技的双引擎
人工智能·python·机器学习
神奇夜光杯1 小时前
Python酷库之旅-第三方库Pandas(202)
开发语言·人工智能·python·excel·pandas·标准库及第三方库·学习与成长
正义的彬彬侠1 小时前
《XGBoost算法的原理推导》12-14决策树复杂度的正则化项 公式解析
人工智能·决策树·机器学习·集成学习·boosting·xgboost
Debroon1 小时前
RuleAlign 规则对齐框架:将医生的诊断规则形式化并注入模型,无需额外人工标注的自动对齐方法
人工智能
羊小猪~~1 小时前
神经网络基础--什么是正向传播??什么是方向传播??
人工智能·pytorch·python·深度学习·神经网络·算法·机器学习
AI小杨1 小时前
【车道线检测】一、传统车道线检测:基于霍夫变换的车道线检测史诗级详细教程
人工智能·opencv·计算机视觉·霍夫变换·车道线检测