【详细】CNN中的卷积计算是什么-实例讲解

本文来自《老饼讲解-BP神经网络》www.bbbdata.com/

卷积计算是CNN中最基本的计算,它是卷积层最重要的组成部分,

下面讲讲CNN的卷积计算过程是什么样的,包括基础卷积计算、多通道卷积计算以及pytorch中的卷积计算。

一、 CNN的基础卷积计算

卷积层是卷积神经网络中最基础的层,基础卷积的计算就是通过一个卷积核对输入进行卷积计算

1.1.一个例子了解CNN的卷积计算是什么

要了解什么是CNN的卷积基础计算,直接通过一个例子来算一算,就一清二楚。 一个卷积计算结果示例如下:

下面详细讲讲上面的卷积结果的具体计算过程:

卷积层中的卷积核就是一个矩阵,直观来看它就是一个窗口,卷积窗口一般为正方形,即长宽一致, 卷积运算通过从左到右,从上往下移动卷积核窗口,将窗口覆盖的每一小块输入进行加权,作为输出

1.2.卷积层的生物意义

卷积核在生物上的意义就相当于动物的眼睛(接受器),

卷积核的大小就相当于眼睛的视野范围(接受野),卷积核的权重就相当于眼睛每一处的接收权重 由于视野有限,所以需要通过逐步移动来查看所有内容,其中眼睛移动的步幅就是卷积的步幅

二、卷积的拓展:多输入通道与多输出通道

2.1.多输入通道卷积

卷积计算支持多通道的输入,当输入为k个通道的时,卷积核为3维矩阵,第3维与输入通道保持一致

2.2.多输出通道卷积

卷积计算也支持多通道输出,需要输出多少个通道,就使用多少个卷积核就可以了

三、卷积的实现

3.1.pytorch实现卷积计算-例子

pytorch中使用torch.nn.Conv2d函数来实现卷积的计算 以上述卷积为例,在pytorch中可以如下实现:

python 复制代码
import torch
X =  torch.tensor([[[[1,3,1,2],[2,6,8,5],[4,2,1,0]]]], dtype=torch.float32)    # 输入数据
c = torch.nn.Conv2d(1, 1, kernel_size=(2,2),bias = False)                      # 初始化卷积类
c.weight.data =torch.tensor([[[[1,2],[2,0]]]], dtype=torch.float32)            # 设置权重 
out = c(X)                                                                     # 对输入进行卷积计算
#-------------打印结果-----------------------
print('输入数据:',X)                 
print('卷积核:',c.weight.data)
print('卷积结果:',out)

运行结果如下:

3.2.Conv2d函数的使用方法

Conv2d函数的完整入参如下:

python 复制代码
torch.nn.Conv2d(in_channels
                ,out_channels
                ,kernel_size
                ,stride       = 1
                ,padding      = 0
                ,dilation     = 1
                ,groups       = 1
                ,bias         = True
                ,padding_mode = 'zeros'
                ,device       = None
                ,dtype        = None)

各个参数的解释如下:

python 复制代码
in_channels   (int) :图象的通道数,也就是决定卷积核的通道数

out_channels (int) :卷积输出的通道数,也就是用多少个卷积核

kernel_size (int/tuple):卷积核的大小,输入5指5*5的卷积核,输入(3,4)指3*4的卷积核

stride (int/tuple):卷积核的步距

padding(int/tuple/str):边缘填充的象素数,5指上下左右都补充5个象素,(3,4)代表上下补充3象素,左右补3象素

padding_mode (str) :象素填充的方式,可选择项- 'zeros', 'reflect', 'replicate','circular'

dilation(int/tuple):"扩张卷积"的专用参数-扩张率,控制kernel各点之间的间隔数量

groups(int):Number of blocked connections from input channels to output channels. Default: 1

bias(bool):是否对卷积结果添加偏置

*详细可进一步查看 《Conv2d的官方说明》


相关链接:

《老饼讲解-神经网络》:老饼讲解-深度学习-通俗易懂

相关推荐
上天夭5 小时前
模型训练篇
人工智能·深度学习·机器学习
Blossom.1185 小时前
AI编译器实战:从零手写算子融合与自动调度系统
人工智能·python·深度学习·机器学习·flask·transformer·tornado
泰迪智能科技016 小时前
分享图书推荐 | 数字图像处理实战
人工智能·深度学习·计算机视觉
Rabbit_QL6 小时前
【深度学习原理】数值稳定性(二):梯度是如何在深度网络中消失与爆炸的
人工智能·深度学习
wanzhong23337 小时前
CUDA学习5-矩阵乘法(共享内存版)
深度学习·学习·算法·cuda·高性能计算
Hcoco_me8 小时前
大模型面试题19:梯度消失&梯度爆炸 纯白话文版
人工智能·rnn·深度学习·自然语言处理·word2vec
是店小二呀9 小时前
在 AtomGit 昇腾 Atlas 800T上解锁 SGLang:零成本打造高性能推理服务
人工智能·pytorch·深度学习·npu
万事可爱^9 小时前
GitCode+昇腾部署Rnj-1模型实践教程
人工智能·深度学习·语言模型·gitcode·本地部署·昇腾npu
高洁019 小时前
图神经网络初探(2)
人工智能·深度学习·算法·机器学习·transformer
祝余Eleanor9 小时前
Day 51 神经网络调参指南
深度学习·神经网络·机器学习