深度学习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()
相关推荐
用户47949283569159 分钟前
每天都在用大模型,但是你知道temperature、top_p、top_k这些常见参数是做什么的吗?
人工智能·面试·llm
z千鑫9 分钟前
【OpenAI】性价比极高的轻量级多模态模型GPT-4.1-mini介绍 + API KEY的使用教程!
人工智能·gpt·ai·语言模型·chatgpt
机器之心12 分钟前
苹果发布会:耳机测心率、手表听音乐、iPhone Air超级薄
人工智能·openai
CoovallyAIHub15 分钟前
基于YOLO集成模型的无人机多光谱风电部件缺陷检测
深度学习·算法·计算机视觉
CoovallyAIHub15 分钟前
几十个像素的小目标,为何难倒无人机?LCW-YOLO让无人机小目标检测不再卡顿
深度学习·算法·计算机视觉
中科岩创17 分钟前
某排水涵洞结构安全自动化监测
人工智能·物联网·自动化
mit6.8241 小时前
[网络入侵AI检测] 模型性能评估与报告
人工智能
黄焖鸡能干四碗1 小时前
智慧教育,智慧校园,智慧安防学校建设解决方案(PPT+WORD)
java·大数据·开发语言·数据库·人工智能
IMER SIMPLE1 小时前
人工智能-python-深度学习-经典网络模型-LeNets5
人工智能·python·深度学习
却道天凉_好个秋1 小时前
深度学习(五):过拟合、欠拟合与代价函数
人工智能·深度学习·过拟合·欠拟合·代价函数