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和Linux系统上的错误
Dec 12 Python
一个基于flask的web应用诞生 组织结构调整(7)
Apr 11 Python
Python跳出多重循环的方法示例
Jul 03 Python
在Python中COM口的调用方法
Jul 03 Python
浅谈django url请求与数据库连接池的共享问题
Aug 29 Python
详解Python图像处理库Pillow常用使用方法
Sep 02 Python
Python TCP通信客户端服务端代码实例
Nov 21 Python
python垃圾回收机制(GC)原理解析
Dec 30 Python
Python turtle画图库&&画姓名实例
Jan 19 Python
Python 生成VOC格式的标签实例
Mar 10 Python
Python 循环读取数据内存不足的解决方案
May 25 Python
Python+OpenCV实现在图像上绘制矩形
Mar 21 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
实现了一个PHP5的getter/setter基类的代码
2007/02/25 PHP
php设计模式 Factory(工厂模式)
2011/06/26 PHP
如何让搜索引擎抓取AJAX内容解决方案
2014/08/25 PHP
php实现的任意进制互转类分享
2015/07/07 PHP
yii2使用ajax返回json的实现方法
2016/05/14 PHP
详解Laravel视图间共享数据与视图Composer
2016/08/04 PHP
原生js 秒表实现代码
2012/07/24 Javascript
jquery 页面滚动到指定DIV实现代码
2013/09/25 Javascript
js实现双击图片放大单击缩小的方法
2015/02/17 Javascript
Bootstrap表单制作代码
2017/03/17 Javascript
详解如何在NodeJS项目中优雅的使用ES6
2017/04/22 NodeJs
Angular2 组件间通过@Input @Output通讯示例
2017/08/24 Javascript
在 Angular-cli 中使用 simple-mock 实现前端开发 API Mock 接口数据模拟功能的方法
2018/11/28 Javascript
vue使用代理解决请求跨域问题详解
2019/07/24 Javascript
jQuery - AJAX load() 实例用法详解
2019/08/27 jQuery
js贪心算法 钱币找零问题代码实例
2019/09/11 Javascript
Vue如何使用混合Mixins和插件开发详解
2020/02/05 Javascript
用Angular实现一个扫雷的游戏示例
2020/05/15 Javascript
解决Vue大括号字符换行踩的坑
2020/11/09 Javascript
在Python操作时间和日期之asctime()方法的使用
2015/05/22 Python
python 实现删除文件或文件夹实例详解
2016/12/04 Python
Python中关键字nonlocal和global的声明与解析
2017/03/12 Python
python3常用的数据清洗方法(小结)
2019/10/31 Python
python xlsxwriter模块的使用
2020/12/24 Python
AmazeUI的下载配置与Helloworld的实现
2020/08/19 HTML / CSS
Herschel Supply Co.美国:背包、手提袋及配件
2020/11/24 全球购物
教师自荐信范文
2013/12/09 职场文书
中英文自我评价常用句型
2013/12/19 职场文书
公司总经理工作职责管理办法
2014/02/28 职场文书
小学开学典礼主持词
2014/03/19 职场文书
2014年村支部书记四风对照检查材料思想汇报
2014/10/02 职场文书
2016年大学生暑期社会实践方案
2015/11/26 职场文书
2016年母亲节广告语
2016/01/28 职场文书
Python通过m3u8文件下载合并ts视频的操作
2021/04/16 Python
Python访问Redis的详细操作
2021/06/26 Python
Python几种酷炫的进度条的方式
2022/04/11 Python