pytorch中的transforms模块实例详解
作者：gaishi_hero 发布时间：2022-04-25 19:50:58
pytorch中的transforms模块中包含了很多种对图像数据进行变换的函数，这些都是在我们进行图像数据读入步骤中必不可少的，下面我们讲解几种最常用的函数，详细的内容还请参考pytorch官方文档（放在文末）。
data_transforms = transforms.Compose([
transforms.RandomResizedCrop(224),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])
从上面的data_transforms变量中我们能够看出进行了多种变换，而Compose方法是将多种变换组合起来。data_transforms中一共包含了四个变换，前两个是对PILImage进行的，分别对其进行随机大小（默认原始图像大小的0.08-1.0）和随机宽高比（默认原始图像宽高比的3/4-4/3）的裁剪，之后resize到指定大小224；以及对原始图像进行随机（默认0.5概率）的水平翻转。
第三个transforms.ToTensor()的变换操作是关键一步，它将PILImage转变为torch.FloatTensor的数据形式，这种数据形式一定是C x H x W的图像格式加上[0,1]的大小范围。它将颜色通道这一维从第三维变换到了第一维。
最后的Normalize变换是对tensor这种数据格式进行的，它的操作是用给定的均值和标准差分别对每个通道的数据进行正则化。具体来说，给定均值(M1,...,Mn)，给定标准差(S1,..,Sn)，其中n是通道数（一般是3），对每个通道进行如下操作：
output[channel] = (input[channel] - mean[channel]) / std[channel]
最后需要强调一点的是，这几个变换的先后顺序有一定的讲究，因为不同的方法所处理的对象不一样，前两种变换是对PILImage进行的，而Normalize则是对tensor进行的，所以处理PILImage的变换方法（大多数方法）都需要放在ToTensor方法之前，而处理tensor的方法（比如Normalize方法）就要放在ToTensor方法之后。
附上pytorch官方参考：https://pytorch.org/docs/stable/torchvision/transforms.html?highlight=torchvision％20transforms
来源：https://blog.csdn.net/gaishi_hero/article/details/81153926
0
投稿猜你喜欢
- 目的：在numpy数组中知道指定元素的索引函数: np.argwhere>>>x>>>array([[0
- python设置环境变量有什么用？1、设置环境变量可以在cmd命令行中找到python命令；2、设置环境变量可以在安装exe的第三包的时候找
- 大家都知道连续的英文或数字能是容器被撑大，不能根据容器的大小自动换行，网页设计初学者可能不知道怎么处理，下面是CSS如何将他们换行的方法！对
- 数据：一个db，2000个表格，2000张表格数据，每条记录30个column，平均每张表格4.3w条记录，总共86388670条记录。机器
一个js用星投票的例子，不是常见的图片版，当然你如果会点js代码的话应该可以改为更加美观的图片投票，原理都一样。本程序只演示了读取星的个数，- 之前用的Python2，连接MySQL用的是MySQLdb。现在换成python3.x了，由于 MySQLdb 模块还不支持 Python3
- 获取Tensor的维数>>> import tensorflow as tf>>> tf.__versi
- K-近邻算法概述简单地说, k-近邻算法采用测量不同特征值之间的距离方法进行分类。k-近邻算法优点：精度高、对异常值不敏感、无数据输入假定。
- SQL Server中包含多种不同类型的页，来满足数据存储的需求。不管是什么类型的页，它们的存储结构都是相同的。每个数据文件都包含相当数量的
- 标量(scalar)数据类型标量(scalar)数据类型没有内部组件，他们大致可分为以下四类：. number. character. da
- 1.准备代码# coding=utf-8class TestDebug: def __init__(self):
- 以下是通过Excel 的VBA连接Oracle并操作Oracle相关数据的示例Excel 通过VBA连接数据库需要安装相应的Oracle客户
- 1。建立数据库表 表名为online 设如下字段 id '用来记录每一个访问都的session.sessionid name
- 前言优化随机森林算法，正确率提高1％~5％（已经有90％+的正确率，再调高会导致过拟合）论文当然是参考的，毕竟出现早的算法都被人研究烂了，什
- Python中pass的作用空语句 do nothing保证格式完整保证语义完整以if语句为例，在c或c++/java中：if(true);
- 1.ACCESS查询(query_info):select * from info where stu = name1 and age =
- 随着对称密码的发展,DES数据加密标准算法由于密钥长度较小(56位),已经不适应当今分布式开放网络对数据加密安全性的要求，因此1997年NI
- # -*- coding: utf-8 -*-import random#数字类class NumberItem: &
- 从事API相关的工作很有挑战性，在高峰期保持系统的稳定及健壮性就是其中之一，这也是我们在Mailgun做很多压力测试的原因。这么久以来，我们
- 基本开发环境· Python 3.6· Pycharm相关模块使用目标网页分析输入想看的小说内容，点击搜索这里会返回很多结果，我只选择第一个
