解析平面卷积/pytorch的nn.Conv2d的计算步骤,in_channels与out_channels如何计算而来

关于二维图像卷积详细过程/pytorch-nn.Conv2d方法

python 复制代码
nn.Conv2d(
  in_channels=inChannel, 
  out_channels=outChannel, 
  kernel_size=kernelSize, 
  stride=stride, 
  padding=padding,
)

具体详细步骤如图

详细说明

  1. 定义outChannel个卷积块,每个卷积块有inChannel个卷积核,所以一共 inChannel * outChannel个卷积核
  2. 输入块的每个通道与卷积核卷积,由此一来,每个卷积块得到一个中间块,共outChannel个块
  3. 每个中间块通道相加,每个中间块压缩成一个平面,共outChannel个平面
  4. 平面叠加起来,成一个输出块,所以输出块的通道为outChannel

其中各方法含义

  • 通道相加 就像把一本书压成一张纸,每个数都相加
  • 通道堆叠 把通道相加后的纸订成一本书,通道合并起来
相关推荐
吃鱼不卡次6 小时前
RT-DETR解码模块(Decoder)
人工智能·深度学习·cross attention·rt-detr·匈牙利匹配·self attention·对比去噪训练
Juchecar6 小时前
假设人类能用光波沟通……
人工智能
K姐研究社6 小时前
AipexBase怎么用?AI 原生BaaS平台一句话做后端开发
人工智能
IT_陈寒6 小时前
SpringBoot 3.2新特性实战:这5个隐藏功能让开发效率翻倍🚀
前端·人工智能·后端
IT_陈寒6 小时前
Vue3性能优化实战:这5个技巧让我的应用加载速度提升70% 🚀
前端·人工智能·后端
Mr.Winter`6 小时前
自动驾驶运动规划 | 基于自行车模型的运动学模型和横向动力学模型详细推导图解
人工智能·机器人·自动驾驶·ros
唯道行6 小时前
计算机图形学·6 OpenGL编程3 谢尔宾斯基垫与三维编程
人工智能·算法·计算机视觉·计算机图形学·三维·谢尔宾斯基垫
落羽的落羽7 小时前
【Linux系统】C/C++的调试器gdb/cgdb,从入门到精通
linux·服务器·c语言·c++·人工智能·学习·机器学习
磊磊落落7 小时前
以自然语言的方式使用 Playwright MCP 进行浏览器自动化操作
人工智能·ai编程