pytorch中的transforms模块实例详解


Posted in Python onDecember 31, 2019

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

以上这篇pytorch中的transforms模块实例详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python3实现连接SQLite数据库的方法
Aug 23 Python
Python常用模块介绍
Nov 21 Python
python压缩文件夹内所有文件为zip文件的方法
Jun 20 Python
使用python实现接口的方法
Jul 07 Python
python设定并获取socket超时时间的方法
Jan 12 Python
判断python对象是否可调用的三种方式及其区别详解
Jan 31 Python
使用python绘制温度变化雷达图
Oct 18 Python
python 轮询执行某函数的2种方式
May 03 Python
python中Ansible模块的Playbook的具体使用
May 28 Python
Python 测试框架unittest和pytest的优劣
Sep 26 Python
详解Python中的进程和线程
Jun 23 Python
Python实现批量将文件复制到新的目录中再修改名称
Apr 12 Python
Pytorch 实现权重初始化
Dec 31 #Python
pytorch 归一化与反归一化实例
Dec 31 #Python
Pytorch 数据加载与数据预处理方式
Dec 31 #Python
pytorch 数据处理:定义自己的数据集合实例
Dec 31 #Python
pytorch: Parameter 的数据结构实例
Dec 31 #Python
Python测试线程应用程序过程解析
Dec 31 #Python
Python TCPServer 多线程多客户端通信的实现
Dec 31 #Python
You might like
用PHP实现文件上传二法
2006/10/09 PHP
处理php自动反斜杠的函数代码
2010/01/05 PHP
PHP实现自动对图片进行滚动显示的方法
2015/03/12 PHP
php通过分类列表产生分类树数组的方法
2015/04/20 PHP
jQuery 扩展对input的一些操作方法
2009/10/30 Javascript
nodejs初步体验篇
2015/11/23 NodeJs
学习JavaScript设计模式之代理模式
2016/01/12 Javascript
JS Attribute属性操作详解
2016/05/19 Javascript
js 递归和定时器的实例解析
2017/02/03 Javascript
JS简单实现点击跳转登陆邮箱功能的方法
2017/10/31 Javascript
剖析Angular Component的源码示例
2018/03/23 Javascript
bootstrap select2插件用ajax来获取和显示数据的实例
2018/08/09 Javascript
在Vant的基础上实现添加表单验证框架的方法示例
2018/12/05 Javascript
7个好用的JavaScript技巧分享(译)
2019/05/07 Javascript
深入探索VueJS Scoped CSS 实现原理
2019/09/23 Javascript
js实现贪吃蛇小游戏(加墙)
2020/07/31 Javascript
[01:47]2018年度DOTA2最佳教练-完美盛典
2018/12/16 DOTA
python操作MySQL数据库具体方法
2013/10/28 Python
python共享引用(多个变量引用)示例代码
2013/12/04 Python
跟老齐学Python之Python安装
2014/09/12 Python
Python探索之实现一个简单的HTTP服务器
2017/10/28 Python
python实现对求解最长回文子串的动态规划算法
2018/06/02 Python
Python SMTP发送邮件遇到的一些问题及解决办法
2018/10/24 Python
celery4+django2定时任务的实现代码
2018/12/23 Python
Html5移动端适配IphoneX等机型的方法
2019/06/25 HTML / CSS
俄罗斯鲜花递送:AMF
2020/04/24 全球购物
助人为乐表扬信范文
2014/01/14 职场文书
调研座谈会发言材料
2014/08/23 职场文书
学生打架检讨书
2014/10/20 职场文书
2015年基层党组织公开承诺书
2015/01/21 职场文书
亮剑观后感600字
2015/06/05 职场文书
小学四年级班主任工作经验交流材料
2015/11/02 职场文书
2019关于实习生工作安排及待遇的管理方案!
2019/07/16 职场文书
导游词之茶卡盐湖
2019/11/26 职场文书
深度学习小工程练习之垃圾分类详解
2021/04/14 Python
一看就懂的MySQL的聚簇索引及聚簇索引是如何长高的
2021/05/25 MySQL