【Pytorch】3.Transforms的运用

什么是Transforms

在PyTorch中,transforms是用于对数据进行预处理、增强和变换的操作集合。transforms通常用于数据载入和训练过程中,可以包括数据的归一化、裁剪、翻转、旋转、缩放等操作,以及将数据转换成PyTorch可以处理的Tensor格式。

Transforms的使用

首先导入包

py 复制代码
from torchvision import transforms

实际上是导入了一个文件,文件名为transforms,我们目前主要查看一下其中的ToTensor

我们可以查看一下其中的源码

我们可以看到这个类的主要作用是 将PIL Image或者是numpy.ndarray类型转化为tensor类型

其中的__call__方法,类似于c++重载()运算符

使用PIL image格式

具体的使用就可以通过

py 复制代码
from PIL import Image
from torchvision import transforms

img_path = "dataset/hymenoptera_data/train/ants/0013035.jpg"
PIL_img = Image.open(img_path)
tensor_tans = transforms.ToTensor()     # 通过transforms中的ToTensor类创建一个对象
img = tensor_tans(PIL_img)     # __call__方法类似于c++中重载了()运算符,我们只需要传入PIL_img格式的图像就可以输出tensor格式的图像

print(img)

我们就可以成功地转化为Tensor格式了

使用numpy.ndarray格式

首先我们要先通过pip安装opencv-python这个库,在终端输入

复制代码
pip install opencv-python

安装成功后导入包

py 复制代码
import cv2

然后使用

py 复制代码
cv_img = cv2.imread(img_path)

创建出来的图片格式就是numpy.ndarray格式

为什么要使用Tensor数据类型

Tensor数据类型包括了我们训练神经网络的一系列参数,以及训练神经网络所需要的格式,这是刚刚我们的格式所不具备的

相关推荐
自律小浮2 小时前
GA3C(GPU/CPU混合式异步优势Actor-Critic)算法实现控制倒立摆
pytorch·python·深度学习·机器学习
weixin_464078075 小时前
Pycharm中Jupyter Notebook 插件常用快捷键
ide·jupyter·pycharm
瑆玥琼6 小时前
机器学习流量识别(pytorch+NSL-KDD+多分类建模)
pytorch·机器学习·分类·语音识别
界面开发小八哥10 小时前
JetBrains IDE v2025.1 升级,AI 智能+语言支持齐飞
ide·人工智能·pycharm·go·jetbrains
何双新10 小时前
Odoo 18进阶开发:打造专业级list,kanban视图Dashboard
python·pycharm
李艺为11 小时前
framework.jar三个jar包的区别
java·pycharm·jar
love530love1 天前
Python 开发环境全栈隔离架构:从 Anaconda 到 PyCharm 的四级防护体系
运维·ide·人工智能·windows·python·架构·pycharm
点云SLAM1 天前
PyTorch 中Tensor常用数据结构(int, list, numpy array等)互相转换和实战示例
数据结构·人工智能·pytorch·算法·list·numpy·tensor
Gyoku Mint1 天前
机器学习×第十二卷:回归树与剪枝策略——她剪去多余的分支,只保留想靠近你的那一层
人工智能·算法·机器学习·数据挖掘·pycharm·回归·剪枝
nvvas1 天前
JETBRAINS IDE 开发环境自定义设置快捷键
pycharm·编辑器·intellij-idea·idea