pytorch中的torch.nn.Conv2d()函数图文详解


Posted in Python onFebruary 28, 2022

一、官方文档介绍

官网

pytorch中的torch.nn.Conv2d()函数图文详解

nn.Conv2d:对由多个输入平面组成的输入信号进行二维卷积

pytorch中的torch.nn.Conv2d()函数图文详解

pytorch中的torch.nn.Conv2d()函数图文详解

二、torch.nn.Conv2d()函数详解

参数详解

torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)

参数 参数类型    
in_channels int Number of channels in the input image 输入图像通道数
out_channels int Number of channels produced by the convolution 卷积产生的通道数
kernel_size (int or tuple) Size of the convolving kernel 卷积核尺寸,可以设为1个int型数或者一个(int, int)型的元组。例如(2,3)是高2宽3卷积核
stride (int or tuple, optional) Stride of the convolution. Default: 1 卷积步长,默认为1。可以设为1个int型数或者一个(int, int)型的元组。
padding (int or tuple, optional) Zero-padding added to both sides of the input. Default: 0 填充操作,控制padding_mode的数目。
padding_mode (string, optional) ‘zeros’, ‘reflect’, ‘replicate’ or ‘circular’. Default: ‘zeros’ padding模式,默认为Zero-padding 。
dilation (int or tuple, optional) Spacing between kernel elements. Default: 1 扩张操作:控制kernel点(卷积核点)的间距,默认值:1。
groups (int, optional) Number of blocked connections from input channels to output channels. Default: 1 group参数的作用是控制分组卷积,默认不分组,为1组。
bias (bool, optional) If True, adds a learnable bias to the output. Default: True 为真,则在输出中添加一个可学习的偏差。默认:True。

参数dilation——扩张卷积(也叫空洞卷积)

dilation操作动图演示如下:

Dilated Convolution with a 3 x 3 kernel and dilation rate 2

扩张卷积核为3×3,扩张率为2

pytorch中的torch.nn.Conv2d()函数图文详解

参数groups——分组卷积

Group Convolution顾名思义,则是对输入feature map进行分组,然后每组分别卷积。

pytorch中的torch.nn.Conv2d()函数图文详解

pytorch中的torch.nn.Conv2d()函数图文详解

三、代码实例

import torch

x = torch.randn(3,1,5,4)
print(x)

conv = torch.nn.Conv2d(1,4,(2,3))
res = conv(x)

print(res.shape)    # torch.Size([3, 4, 4, 2])

输入:x[ batch_size, channels, height_1, width_1 ]

  • batch_size,一个batch中样本的个数 3
  • channels,通道数,也就是当前层的深度 1
  • height_1, 图片的高 5
  • width_1, 图片的宽 4

卷积操作:Conv2d[ channels, output, height_2, width_2 ]

  • channels,通道数,和上面保持一致,也就是当前层的深度 1
  • output ,输出的深度 4【需要4个filter】
  • height_2,卷积核的高 2
  • width_2,卷积核的宽 3

输出:res[ batch_size,output, height_3, width_3 ]

  • batch_size,,一个batch中样例的个数,同上 3
  • output, 输出的深度 4
  • height_3, 卷积结果的高度 4
  • width_3,卷积结果的宽度 2

一个样本卷积示例:

pytorch中的torch.nn.Conv2d()函数图文详解

pytorch中的torch.nn.Conv2d()函数图文详解

总结 

到此这篇关于pytorch中torch.nn.Conv2d()函数的文章就介绍到这了,更多相关pytorch torch.nn.Conv2d()函数内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
深入学习Python中的装饰器使用
Jun 20 Python
Python列表删除的三种方法代码分享
Oct 31 Python
python中文分词教程之前向最大正向匹配算法详解
Nov 02 Python
django 修改server端口号的方法
May 14 Python
Python smtplib实现发送邮件功能
May 22 Python
python自带tkinter库实现棋盘覆盖图形界面
Jul 17 Python
Python 3.8 新功能全解
Jul 25 Python
python实现实时视频流播放代码实例
Jan 11 Python
Python flask框架实现查询数据库并显示数据
Jun 04 Python
Python接收手机短信的代码整理
Aug 02 Python
如何利用Python动态模拟太阳系运转
Sep 04 Python
python分分钟绘制精美地图海报
Feb 15 Python
python3中apply函数和lambda函数的使用详解
Feb 28 #Python
你需要掌握的20个Python常用技巧
Feb 28 #Python
python opencv将多个图放在一个窗口的实例详解
pandas中关于apply+lambda的应用
Feb 28 #Python
Python中的datetime包与time包包和模块详情
Feb 28 #Python
Python 数据可视化神器Pyecharts绘制图像练习
Python使用OpenCV实现虚拟缩放效果
You might like
分享一个PHP数据流应用的简单例子
2012/06/01 PHP
php实现的click captcha点击验证码类实例
2014/09/23 PHP
php实现单笔转账到支付宝功能
2018/10/09 PHP
javascript 读取XML数据,在页面中展现、编辑、保存的实现
2009/10/27 Javascript
Node.js和PHP根据ip获取地理位置的方法
2014/03/14 Javascript
JavaScript获取table中某一列的值的方法
2014/05/06 Javascript
iframe如何动态创建及释放其所占内存
2014/09/03 Javascript
jQuery实现仿淘宝带有指示条的图片转动切换效果完整实例
2015/03/04 Javascript
JavaScript中return用法示例
2016/11/29 Javascript
jQuery设置和获取select、checkbox、radio的选中值方法
2017/01/01 Javascript
bootstrap multiselect 多选功能实现方法
2017/06/05 Javascript
Vue 3.0双向绑定原理的实现方法
2019/10/23 Javascript
使用Node.js实现base64和png文件相互转换的方法
2020/03/11 Javascript
Vue脚手架编写试卷页面功能
2020/03/17 Javascript
python结合shell查询google关键词排名的实现代码
2016/02/27 Python
python用装饰器自动注册Tornado路由详解
2017/02/14 Python
Python标准库inspect的具体使用方法
2017/12/06 Python
pandas实现选取特定索引的行
2018/04/20 Python
python实现图片批量压缩程序
2018/07/23 Python
使用django-guardian实现django-admin的行级权限控制的方法
2018/10/30 Python
Python箱型图绘制与特征值获取过程解析
2019/10/22 Python
python使用配置文件过程详解
2019/12/28 Python
Python把图片转化为pdf代码实例
2020/07/28 Python
使用豆瓣源来安装python中的第三方库方法
2021/01/26 Python
使用placeholder属性设置input文本框的提示信息
2020/02/19 HTML / CSS
国贸专业个人求职信范文
2014/01/08 职场文书
甜美蛋糕店创业计划书
2014/01/30 职场文书
人事行政经理岗位职责
2014/06/18 职场文书
法学专业求职信
2014/07/15 职场文书
办理信用卡收入证明范例
2014/09/13 职场文书
加强机关作风建设心得体会
2014/10/22 职场文书
毕业生班级鉴定评语
2015/01/04 职场文书
开幕式邀请函
2015/01/31 职场文书
2015年计划生育协会工作总结
2015/05/13 职场文书
居委会工作总结2015
2015/05/18 职场文书
python 安全地删除列表元素的方法
2022/03/16 Python