pytorch1.0中torch.nn.Conv2d用法详解


Posted in Python onJanuary 10, 2020

Conv2d的简单使用

torch 包 nn 中 Conv2d 的用法与 tensorflow 中类似,但不完全一样。

在 torch 中,Conv2d 有几个基本的参数,分别是

in_channels 输入图像的深度

out_channels 输出图像的深度

kernel_size 卷积核大小,正方形卷积只为单个数字

stride 卷积步长,默认为1

padding 卷积是否造成尺寸丢失,1为不丢失

与tensorflow不一样的是,pytorch中的使用更加清晰化,我们可以使用这种方法定义输入与输出图像的深度并同时指定使用的卷积核的大小。

而我们的输入则由经由 Conv2d 定义的参数传入,如下所示:

# 定义一个输入深度为1,输出为6,卷积核大小为 3*3 的 conv1 变量
self.conv1 = nn.Conv2d(1, 6, 3)
# 传入原始输入x,以获得长宽与x相当,深度为6的卷积部分
x = self.conv1(x)

要注意的是,Conv2d中所需要的输入顺序为

batchsize, nChannels, Height, Width

其他的简单使用

同样的,与 Conv2d 类似的函数还有很多,类似 max_pool2d、relu等等,他们的使用方法与 Conv2d 极为类似,如下所示:

# relu函数的使用
F.relu(self.conv1(x))
# 池化函数的使用
F.max_pool2d(F.relu(self.conv2(x)), 2)

以上这篇pytorch1.0中torch.nn.Conv2d用法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
介绍Python中的一些高级编程技巧
Apr 02 Python
实例讲解Python设计模式编程之工厂方法模式的使用
Mar 02 Python
python虚拟环境virtualenv的安装与使用
Sep 21 Python
python实现批量按比例缩放图片效果
Mar 30 Python
python basemap 画出经纬度并标定的实例
Jul 09 Python
Python异常模块traceback用法实例分析
Oct 22 Python
pygame库实现移动底座弹球小游戏
Apr 14 Python
opencv python Canny边缘提取实现过程解析
Feb 03 Python
基于python检查矩阵计算结果
May 21 Python
Django实现前台上传并显示图片功能
May 29 Python
sqlalchemy实现时间列自动更新教程
Sep 02 Python
python工具——Mimesis的简单使用教程
Jan 16 Python
pytorch 利用lstm做mnist手写数字识别分类的实例
Jan 10 #Python
Tensorflow Summary用法学习笔记
Jan 10 #Python
TENSORFLOW变量作用域(VARIABLE SCOPE)
Jan 10 #Python
python numpy数组复制使用实例解析
Jan 10 #Python
关于Pytorch的MNIST数据集的预处理详解
Jan 10 #Python
详解pycharm连接不上mysql数据库的解决办法
Jan 10 #Python
pycharm双击无响应(打不开问题解决办法)
Jan 10 #Python
You might like
php绘制一条弧线的方法
2015/01/24 PHP
Aster vs Newbee BO5 第二场2.19
2021/03/10 DOTA
Js基础学习资料
2010/11/23 Javascript
extjs3 combobox取value和text案例详解
2013/02/06 Javascript
js判断生效时间不得大于失效时间的思路及代码
2013/04/23 Javascript
jQuery+html5实现div弹出层并遮罩背景
2015/04/15 Javascript
详解Vue 实例中的生命周期钩子
2017/03/21 Javascript
JS设计模式之单例模式(一)
2017/09/29 Javascript
基于Datatables跳转到指定页的简单实例
2017/11/09 Javascript
详解基于vue的服务端渲染框架NUXT
2018/06/20 Javascript
VUE预渲染及遇到的坑
2018/09/03 Javascript
ios设备中angularjs无法改变页面title的解决方法
2018/09/13 Javascript
Puppet的一些技巧
2018/09/17 Javascript
js实现拖动缓动效果
2020/01/13 Javascript
[01:27]2014DOTA2展望TI 剑指西雅图IG战队专访
2014/06/30 DOTA
[02:17]2016国际邀请赛中国区预选赛VG战队领队采访
2016/06/26 DOTA
跟老齐学Python之深入变量和引用对象
2014/09/24 Python
python如何在终端里面显示一张图片
2016/08/17 Python
python通过百度地图API获取某地址的经纬度详解
2018/01/28 Python
详解Django的model查询操作与查询性能优化
2018/10/16 Python
python list格式数据excel导出方法
2018/10/31 Python
Python机器学习算法库scikit-learn学习之决策树实现方法详解
2019/07/04 Python
详解Django 时间与时区设置问题
2019/07/23 Python
python实现扫雷游戏
2020/03/03 Python
Numpy 多维数据数组的实现
2020/06/18 Python
详解CSS3浏览器兼容
2016/12/14 HTML / CSS
俄罗斯在线手表和珠宝商店:AllTime
2019/09/28 全球购物
酒店保安领班职务说明书
2014/03/04 职场文书
各营销点岗位职责范本
2014/03/05 职场文书
音乐教育感言
2014/03/05 职场文书
行政办公室岗位职责
2014/03/18 职场文书
领导班子四风问题对照检查材料
2014/09/27 职场文书
优秀教师先进材料
2014/12/16 职场文书
新手,如何业余时间安排好写作、提高写作能力?
2019/10/21 职场文书
详解java如何集成swagger组件
2021/06/21 Java/Android
基于Python编写一个监控CPU的应用系统
2022/06/25 Python