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 相关文章推荐
python获得图片base64编码示例
Jan 16 Python
python实现调用其他python脚本的方法
Oct 05 Python
Python实现Mysql数据库连接池实例详解
Apr 11 Python
Python callable()函数用法实例分析
Mar 17 Python
Numpy数据类型转换astype,dtype的方法
Jun 09 Python
值得收藏,Python 开发中的高级技巧
Nov 23 Python
Python使用pymongo库操作MongoDB数据库的方法实例
Feb 22 Python
解决Django中修改js css文件但浏览器无法及时与之改变的问题
Aug 31 Python
python3.x 生成3维随机数组实例
Nov 28 Python
python通过对字典的排序,对json字段进行排序的实例
Feb 27 Python
python 爬虫请求模块requests详解
Dec 04 Python
python 如何用urllib与服务端交互(发送和接收数据)
Mar 04 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模板Smarty的初级使用方法以及心得分享
2013/06/21 PHP
PHP中preg_match函数正则匹配的字符串长度问题
2015/05/27 PHP
ThinkPHP实现更新数据实例详解(demo)
2016/06/29 PHP
PHP7新特性foreach 修改示例介绍
2016/08/26 PHP
PHP自定义错误用法示例
2016/09/28 PHP
php 微信公众平台开发模式实现多客服的实例代码
2016/11/07 PHP
Mootools 1.2教程 滚动条(Slider)
2009/09/15 Javascript
jquery validator 插件增加日期比较方法
2010/02/21 Javascript
js ondocumentready onmouseover onclick onmouseout 样式
2010/07/22 Javascript
Jquery异步请求数据实例代码
2011/12/28 Javascript
鼠标经过显示二级菜单js特效
2013/08/13 Javascript
javascript 闭包详解
2015/02/15 Javascript
JavaScript字符串常用的方法
2016/03/10 Javascript
jQuery+ajax简单实现文件上传的方法
2016/06/03 Javascript
js判断请求的url是否可访问,支持跨域判断的实现方法
2016/09/17 Javascript
Jquery Easyui搜索框组件SearchBox使用详解(19)
2016/12/17 Javascript
JS敏感词过滤代码
2016/12/23 Javascript
微信小程序实现锚点定位楼层跳跃的实例
2017/05/18 Javascript
详解vue.js数据传递以及数据分发slot
2018/01/20 Javascript
Jquery的autocomplete插件用法及参数讲解
2019/03/12 jQuery
JavaScript中this的全面解析及常见实例
2019/05/14 Javascript
微信小程序scroll-view点击项自动居中效果的实现
2020/03/25 Javascript
webpack 如何同时输出压缩和未压缩的文件的实现步骤
2020/06/05 Javascript
python服务器端收发请求的实现代码
2014/09/29 Python
python3操作mysql数据库的方法
2017/06/23 Python
python删除字符串中指定字符的方法
2018/08/13 Python
python创造虚拟环境方法总结
2019/03/04 Python
opencv 获取rtsp流媒体视频的实现方法
2019/08/23 Python
使用python+poco+夜神模拟器进行自动化测试实例
2020/04/23 Python
python用tkinter实现一个gui的翻译工具
2020/10/26 Python
介绍一下javax.servlet.Servlet接口及其主要方法
2015/11/30 面试题
应届生个人求职信模板
2013/11/26 职场文书
优秀辅导员事迹材料
2014/02/16 职场文书
群众路线教育实践活动个人对照检查材料思想汇报(社区班子)
2014/10/06 职场文书
解决Nginx 配置 proxy_pass 后 返回404问题
2021/03/31 Servers