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保存字符串到文件的方法
Jul 01 Python
Python学习教程之常用的内置函数大全
Jul 14 Python
详解python中自定义超时异常的几种方法
Jul 29 Python
Python-numpy实现灰度图像的分块和合并方式
Jan 09 Python
pytorch方法测试详解——归一化(BatchNorm2d)
Jan 15 Python
Python matplotlib修改默认字体的操作
Mar 05 Python
tensorflow常用函数API介绍
Apr 19 Python
python定义类的简单用法
Jul 24 Python
Django生成数据库及添加用户报错解决方案
Oct 09 Python
如何基于matlab相机标定导出xml文件
Nov 02 Python
python实现简单区块链结构
Apr 25 Python
python实现会员信息管理系统(List)
Mar 18 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
浅析Dos下运行php.exe,出现没有找到php_mbstring.dll 错误的解决方法
2013/06/29 PHP
php遍历目录输出目录及其下的所有文件示例
2014/01/27 PHP
php微信公众开发之获取周边酒店信息的方法
2014/12/22 PHP
div层的移动及性能优化
2010/11/16 Javascript
js中将HTMLCollection/NodeList/伪数组转换成数组的代码
2011/07/31 Javascript
BootStrop前端框架入门教程详解
2016/12/25 Javascript
基于ES6 Array.of的用法(实例讲解)
2017/09/05 Javascript
分析JS单线程异步io回调的特性
2017/12/01 Javascript
微信小程序实现图片上传功能实例(前端+PHP后端)
2018/01/10 Javascript
微信小程序实现人脸识别
2018/05/25 Javascript
JS使用对象的defineProperty进行变量监控操作示例
2019/02/02 Javascript
Vue 实现前端权限控制的示例代码
2019/07/09 Javascript
JavaScript设计模式之门面模式原理与实现方法分析
2020/03/09 Javascript
原生JavaScript写出Tabs标签页的实例代码
2020/07/20 Javascript
[39:08]完美世界DOTA2联赛PWL S3 LBZS vs CPG 第一场 12.12
2020/12/16 DOTA
Python爬虫框架Scrapy常用命令总结
2018/07/26 Python
python学习之hook钩子的原理和使用
2018/10/25 Python
使用 Python 玩转 GitHub 的贡献板(推荐)
2019/04/04 Python
python进程的状态、创建及使用方法详解
2019/12/06 Python
pytorch 批次遍历数据集打印数据的例子
2019/12/30 Python
aws 通过boto3 python脚本打pach的实现方法
2020/05/10 Python
python如何调用字典的key
2020/05/25 Python
安装python3.7编译器后如何正确安装opnecv的方法详解
2020/06/16 Python
Python如何给你的程序做性能测试
2020/07/29 Python
如何在scrapy中集成selenium爬取网页的方法
2020/11/18 Python
GUESS西班牙官方网上商城:美国服饰品牌
2017/03/15 全球购物
Nike爱尔兰官方网站:Nike.com (IE)
2018/03/12 全球购物
竞聘医务工作人员的自我评价分享
2013/11/04 职场文书
大学生全国两会报告感想
2014/03/17 职场文书
三问三解心得体会
2014/09/05 职场文书
入党积极分子十八届四中全会思想汇报
2014/10/23 职场文书
教师创先争优承诺书
2015/04/27 职场文书
2015年清剿火患专项行动工作总结
2015/07/27 职场文书
2019单位介绍信怎么写
2019/06/24 职场文书
中国现代文学之经典散文三篇
2019/09/18 职场文书
教你怎么用Python生成九宫格照片
2021/05/20 Python