📜  卷积的概念(1)

📅  最后修改于: 2023-12-03 15:37:04.414000             🧑  作者: Mango

卷积的概念

卷积是信号处理和图像处理中一种基本的操作,其用途包括滤波、特征提取、图像压缩等。在深度学习中,卷积神经网络(Convolutional Neural Networks, CNNs)用卷积操作处理图像、语音、文本等数据,取得了很大成功。

1. 定义

卷积是一种将两个函数叠加在一起,生成第三个函数的数学操作。卷积运算通常表示为 $\ast$ 或 $*$。对于两个函数 $f$ 和 $g$,它们的卷积函数 $h$ 定义为:

$$(f * g)(t) = \int_{-\infty}^{\infty} f(\tau) g(t - \tau) d\tau$$

当 $f$ 和 $g$ 都是离散函数时,卷积的定义为:

$$(f * g)[n] = \sum_{m=-\infty}^{\infty} f[m] g[n - m]$$

其中,$n$ 是整数,$f[n]$ 和 $g[n]$ 表示两个离散函数在 $n$ 处的值。

2. 图像处理中的卷积

在图像处理中,卷积是一种将图像和一个卷积核(或滤波器)叠加在一起,生成一个新的图像的数学操作。卷积核通常是一个小的二维矩阵,它在图像上进行滑动,每个位置都计算出一个值,最终生成的图像就是这些值组成的矩阵。

对于一个输入图像 $I$ 和一个卷积核 $K$,它们的卷积结果 $O$ 可以表示为:

$$O_{i,j} = \sum_{m} \sum_{n} I_{m,n} K_{i-m,j-n}$$

其中,$i$ 和 $j$ 分别表示输出图像的行和列,$m$ 和 $n$ 分别表示输入图像的行和列。这个式子表明,输出图像的每个像素都是输入图像和卷积核之间的相似度的一种测量,相当于一种特征提取操作。

3. CNN中的卷积

在CNN中,卷积层使用卷积操作处理输入张量(通常是一个图像),并通过一些卷积核输出一些特征图(通常是一些2D矩阵)。这些特征图可以被下一层的卷积层或其他类型的层继续使用。

使用卷积操作的好处在于它可以共享权重(权值共享),使得CNN的计算效率得到提高。此外,卷积操作还可以保留一定的平移不变性,使得CNN在处理图像时不会受到位置变化的干扰。

下面是PyTorch中使用卷积层的一个简单例子:

import torch
import torch.nn as nn

# 创建一个大小为 (N, C_{in}, H_{in}, W_{in}) 的输入张量
x = torch.randn(1, 3, 224, 224)

# 创建一个卷积层,输入通道数为 3,输出通道数为 64,卷积核大小为 3x3,步长为 1,填充为 1
conv = nn.Conv2d(3, 64, kernel_size=3, stride=1, padding=1)

# 对输入张量进行卷积操作,得到一个大小为 (N, C_{out}, H_{out}, W_{out}) 的输出张量
out = conv(x)
4. 小结

卷积是一种基本的信号处理和图像处理操作,也是CNN的核心操作。在图像处理中,卷积可以用于图像滤波、特征提取等任务;在CNN中,卷积层可以生成特征图并进行权值共享,从而提高计算效率。