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实现将n个点均匀地分布在球面上的方法
Mar 12 Python
Python中利用函数装饰器实现备忘功能
Mar 30 Python
Python标准库之Sys模块使用详解
May 23 Python
python 读取文件并替换字段的实例
Jul 12 Python
Python实现快速傅里叶变换的方法(FFT)
Jul 21 Python
Python创建一个空的dataframe,并循环赋值的方法
Nov 08 Python
使用python3调用wxpy模块监控linux日志并定时发送消息给群组或好友
Jun 05 Python
OpenCV模板匹配matchTemplate的实现
Oct 18 Python
python 操作hive pyhs2方式
Dec 21 Python
Scrapy框架实现的登录网站操作示例
Feb 06 Python
Python urllib.request对象案例解析
May 11 Python
PyTorch预训练Bert模型的示例
Nov 17 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设计模式 Template (模板模式)
2011/06/26 PHP
php获得url参数中具有&的值的方法
2014/03/05 PHP
php中多维数组按指定value排序的实现代码
2014/08/19 PHP
PHP使用ODBC连接数据库的方法
2015/07/18 PHP
基于jquery的inputlimiter 实现字数限制功能
2010/05/30 Javascript
caller和callee的区别介绍及演示结果
2013/03/10 Javascript
javascript setTimeout和setInterval计时的区别详解
2013/06/21 Javascript
jQuery的控件及事件(输入控件及回车事件)使用示例
2013/07/25 Javascript
Node.js中使用mongoskin操作mongoDB实例
2014/09/28 Javascript
Javascript优化技巧之短路表达式详细介绍
2015/03/27 Javascript
JavaScript Function函数类型介绍
2015/04/08 Javascript
webpack中CommonsChunkPlugin详细教程(小结)
2017/11/09 Javascript
解决vue路由后界面没有变化,但是链接有的问题
2018/09/01 Javascript
Python实现统计英文单词个数及字符串分割代码
2015/05/28 Python
Python实现程序的单一实例用法分析
2015/06/03 Python
Python的Flask站点中集成xhEditor文本编辑器的教程
2016/06/13 Python
python实现字典(dict)和字符串(string)的相互转换方法
2017/03/01 Python
使用Python3制作TCP端口扫描器
2017/04/17 Python
Python实现的生成格雷码功能示例
2018/01/24 Python
Python单元测试实例详解
2018/05/25 Python
python爬虫实例详解
2018/06/19 Python
python tkinter的消息框模块(messagebox,simpledialog)
2020/11/07 Python
python字符串拼接+和join的区别详解
2020/12/03 Python
美国医生配方营养补充剂供应商:Healthy Directions
2019/07/10 全球购物
法国春天百货官网:Printemps.com
2020/06/29 全球购物
有abstract方法的类一定要用abstract修饰吗
2016/03/14 面试题
法务专员岗位职责
2014/01/02 职场文书
班训口号大全
2014/06/18 职场文书
倡议书格式
2014/08/30 职场文书
授权委托书(法人单位用)
2014/09/29 职场文书
2014年行政人事工作总结
2014/12/09 职场文书
2014年中职班主任工作总结
2014/12/16 职场文书
人口与计划生育责任书
2015/05/09 职场文书
如何设计高效合理的MySQL查询语句
2021/05/26 MySQL
教你用python实现12306余票查询
2021/06/30 Python
Python实现查询剪贴板自动匹配信息的思路详解
2021/07/09 Python