动手学深度学习(pytorch土堆)-03Transforms简单入门学习

1

torchvision中的transforms主要是对图片进行一些变换。

transforms结构及用法

将特定格式图片经过transforms里面的工具处理输出预期的图像

Totensor使用

tensor数据类型可以理解为包装了一些反向神经网络所需要的一些参数

PIL_Image读取

c 复制代码
img_path="hymenoptera_data/train/ants/0013035.jpg"
img=Image.open(img_path)
# print(img)
tensor_trans=ToTensor()
tensor_img=tensor_trans(img)#将img图片转换成tensor数据类型的图片
# print(tensor_img)

numpy_ndarray读取

使用opencv读取

c 复制代码
cv_img=cv2.imread(img_path)
print(type(cv_img))

结果<class 'numpy.ndarray'>

与前面结合

c 复制代码
from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
from torchvision.transforms import ToTensor
import cv2
#通过transforms.ToTensor去看两个问题
#1.transforms该怎么使用
#2.为什么我们需要Tensor数据类型
#绝对路径E:\学习\研究生阶段\python-learning\pytorch\hymenoptera_data\train\ants\0013035.jpg
#相对路径hymenoptera_data/train/ants/0013035.jpg
img_path="hymenoptera_data/train/ants/0013035.jpg"
img=Image.open(img_path)
writer=SummaryWriter("logs")

# print(img)
tensor_trans=ToTensor()
tensor_img=tensor_trans(img)#将img图片转换成tensor数据类型的图片
# print(tensor_img)
writer.add_image("tensor_img",tensor_img)
#tensorboard --logdir=logs
writer.close()

# cv_img=cv2.imread(img_path)
# print(type(cv_img))
相关推荐
酌量2 分钟前
路径平滑优化详解(二次规划): 数学建模与目标函数推导
经验分享·笔记·学习·机器人·自动驾驶
棱镜研途25 分钟前
科研快报 |声波“听”见火灾温度:混合深度学习重构三维温度场
人工智能·深度学习·目标检测·重构·传感·声波测温·火灾安全
ygyqinghuan35 分钟前
PyTorch 实现 MNIST 手写数字识别
人工智能·pytorch·python
2401_8369003338 分钟前
PyTorch实现MNIST手写数字识别:从数据到模型全解析
pytorch·深度学习·可视化训练
武子康39 分钟前
AI-调查研究-102-具身智能 智能机械臂、自动驾驶与人形机器人的模仿学习、强化学习与多模态融合趋势
人工智能·深度学习·机器学习·ai·机器人·强化学习·具身智能
史锦彪1 小时前
用 PyTorch 实现 MNIST 手写数字识别:从入门到实践
人工智能·pytorch·python
董建光d1 小时前
PyTorch 实现 MNIST 手写数字识别完整流程(含数据处理、模型构建与训练可视化)
人工智能·pytorch·python
爱倒腾的老唐2 小时前
01、如何学习单片机
单片机·嵌入式硬件·学习
于小汐在咯8 小时前
词根学习笔记 | Agri系列
笔记·学习
霜绛8 小时前
Unity:Json笔记——Json文件格式、JsonUtlity序列化和反序列化
学习·unity·json·游戏引擎