(六)卷积神经网络:深度学习在计算机视觉中的应用

发布于:2025-06-09 ⋅ 阅读:(15) ⋅ 点赞:(0)

卷积神经网络(Convolutional Neural Network,CNN)是一类专门用于处理具有网格结构数据(如图像、音频)的深度学习模型。CNN 在图像识别、视频分析等领域取得了巨大成功,其独特的架构设计使其能够有效地提取数据中的空间和时间相关性。

1 从全连接层到卷积层

在神经网络的发展过程中,全连接层(Fully Connected Layer)是最基本的组成部分。然而,随着人们对图像处理需求的增加,全连接层逐渐显现出一些局限性。卷积层(Convolutional Layer)应运而生,为图像处理等任务提供了更高效、更有效的解决方案。

1.1 全连接层的局限性

全连接层中,每个神经元与前一层的所有神经元相连,适用于处理特征维度不大的数据。但图像数据通常具有高维度,直接使用全连接层处理会导致以下问题:

  • 参数量巨大:假设输入图像大小为 28 × 28 28 \times 28 28×28(如MNIST数据集),若隐藏层有256个神经元,仅这一层的参数量(权重)就达 28 × 28 × 256 = 200 , 704 28 \times 28 \times 256 = 200,704 28×28×256=200,704,模型容易过拟合且训练效率低。
  • 未考虑空间结构信息:全连接层未考虑图像像素间的空间关系,不利于提取图像局部特征。
1.2 卷积层的优势

卷积层通过局部连接和参数共享机制克服了全连接层的局限性:

  • 局部连接:卷积核(Filter)仅与输入数据的一个局部区域连接,适合捕捉图像局部特征,如边缘、纹理。
  • 参数共享:同一个卷积核在输入数据的不同位置共享参数,减少参数量并使模型对平移具有不变性。

假设卷积核大小为 3 × 3 3 \times 3 3×3,输出通道数为 1 1 1,输入图像大小为 28 × 28 28 \times 28 28×28,此时卷积层的参数量(权重)仅为 3 × 3 × 1 = 9 3 \times 3 \times 1 = 9 3×3×1=9(不考虑偏置项),大幅降低了参数量。

1.3 卷积层的工作原理

卷积层的核心操作是卷积运算,即卷积核在输入数据上滑动,与局部区域进行逐元素乘积并求和,生成卷积特征图(Feature Map)。这个过程可以提取输入数据的局部特征。

  • 卷积核:一个小的二维矩阵(如 3 × 3 3 \times 3 3×3),包含可学习的参数,用于提取局部特征。
  • 特征图:卷积运算的结果,表示输入数据的局部特征响应。
1.4 卷积层的结构

卷积层的结构包括输入通道、卷积核、输出通道等。对于彩色图像(如RGB格式),有3个输入通道,每个卷积核对应3个输入通道的权重。卷积运算分别对每个输入通道进行,结果相加得到一个输出通道。使用多个卷积核可以生成多个输出通道,每个卷积核提取不同的特征。

1.5 公式表达

假设输入数据为 X ∈ R H × W × C i n X \in \mathbb{R}^{H \times W \times C_{in}} XRH×W×Cin(高度 H H H、宽度 W W W、输入通道数 C i n C_{in} Cin),卷积核大小为 K × K K \times K K×K,输出通道数为 C o u t C_{out} Cout,则卷积层的输出数据为 Y ∈ R H o u t × W o u t × C o u t Y \in \mathbb{R}^{H_{out} \times W_{out} \times C_{out}} YRHout×Wout×Cout,其中:

H o u t = ⌊ H + 2 P − K S ⌋ + 1 H_{out} = \left\lfloor \frac{H + 2P - K}{S} \right\rfloor + 1 Hout=SH+2PK+1

W o u t = ⌊ W + 2 P − K S ⌋ + 1 W_{out} = \left\lfloor \frac{W + 2P - K}{S} \right\rfloor + 1 Wout=SW+2PK+1

  • P P P 是填充(padding)大小。
  • S S S 是步幅(stride)大小。

示例代码

以下是使用PyTorch实现的全连接层和卷积层对比示例:

import torch
import torch.nn as nn

# 定义全连接层
fc_layer = nn.Linear(28 * 28, 256)  # 输入维度28*28,输出维度256
print(f"全连接层的参数量:{fc_layer.weight.numel()}")

# 定义卷积层
conv_layer = nn.Conv2d(1, 1, kernel_size=3, padding=1)  # 输入通道1,输出通道1,卷积核大小3
print(f"卷积层的参数量:{conv_layer.weight.numel()}")

# 假设输入图像为28x28
input_image = torch.randn(1, 1, 28, 28)  # 批量大小1,输入通道1,高度28,宽度28

# 计算全连接层输出
fc_input = input_image.view(-1, 28 * 28)  # 将图像展平为784维向量
fc_output = fc_layer(fc_input)
print(f"全连接层输出维度:{fc_output.shape}")

# 计算卷积层输出
conv_output = conv_layer(input_image)
print(f"卷积层输出维度:{conv_output.shape}")

在这个示例中,我们定义了一个全连接层和一个卷积层,并计算了它们的参数量和输出维度。可以看到,卷积层的参数量远少于全连接层,同时卷积层能够保持输入图像的空间结构信息。

2 图像卷积

图像卷积是卷积神经网络(CNN)的核心操作之一,通过卷积核在图像上滑动,提取图像的局部特征。这一节将详细讲解图像卷积的概念、操作原理以及在PyTorch中的实现。

2.1 卷积核与特征图
  • 卷积核(Filter):一个小的二维矩阵,包含可学习的参数,用于提取图像的局部特征。卷积核在图像上滑动,与图像的局部区域进行逐元素乘积并求和,生成卷积特征图。
  • 特征图(Feature Map):卷积操作的结果,表示图像的局部特征响应。每个特征图对应一个卷积核,多个卷积核可以生成多个特征图。
2.2 卷积操作的数学原理

假设输入图像 X ∈ R H × W X \in \mathbb{R}^{H \times W} XRH×W,卷积核 K ∈ R K × K K \in \mathbb{R}^{K \times K} KRK×K,卷积操作的输出特征图 Y ∈ R H out × W out Y \in \mathbb{R}^{H_{\text{out}} \times W_{\text{out}}} YRHout×Wout,其中:

Y i , j = ∑ m = 0 K − 1 ∑ n = 0 K − 1 X i + m , j + n ⋅ K m , n Y_{i,j} = \sum_{m=0}^{K-1} \sum_{n=0}^{K-1} X_{i+m, j+n} \cdot K_{m,n} Yi,j=m=0K1n=0K1Xi+m,j+nKm,n

  • H H H W W W 是输入图像的高度和宽度。
  • K K K 是卷积核的大小。
  • H out H_{\text{out}} Hout W out W_{\text{out}} Wout是输出特征图的高度和宽度,计算公式为:
    H out = ⌊ H + 2 P − K S ⌋ + 1 H_{\text{out}} = \left\lfloor \frac{H + 2P - K}{S} \right\rfloor + 1 Hout=SH+2PK+1
    W out = ⌊ W + 2 P − K S ⌋ + 1 W_{\text{out}} = \left\lfloor \frac{W + 2P - K}{S} \right\rfloor + 1 Wout=SW+2PK+1
    • P P P 是填充(Padding)大小。
    • S S S 是步幅(Stride)大小。
2.3 填充(Padding)和步幅(Stride)
  • 填充(Padding):在图像边缘添加额外的像素(通常是零),用于控制输出特征图的尺寸。常见的填充方式有 VALID(无填充)和 SAME(保持输出特征图尺寸与输入图像相同)。
  • 步幅(Stride):卷积核在图像上滑动的步长。较大的步幅会减少输出特征图的尺寸。
2.4 卷积操作的代码实现

单通道卷积

import torch
import torch.nn as nn

# 定义输入图像和卷积核
input_image = torch.tensor([[1.0, 2.0, 3.0, 4.0],
                           [5.0, 6.0, 7.0, 8.0],
                           [9.0, 10.0, 11.0, 12.0],
                           [13.0, 14.0, 15.0, 16.0]])

conv_kernel = torch.tensor([[1.0, 0.0],
                            [0.0, -1.0]])

# 将输入图像和卷积核转换为PyTorch张量
input_tensor = input_image.unsqueeze(0).unsqueeze(0)  # 添加批量维度和通道维度
kernel_tensor = conv_kernel.unsqueeze(0).unsqueeze(0)  # 添加输入通道和输出通道维度

# 定义卷积层
conv_layer = nn.Conv2d(in_channels=1, out_channels=1, kernel_size=2, stride=1, padding=0, bias=False)
conv_layer.weight.data = kernel_tensor

# 执行卷积操作
output = conv_layer(input_tensor)

# 打印输入图像、卷积核和输出特征图
print("输入图像:")
print(input_image)
print("\n卷积核:")
print(conv_kernel)
print("\n输出特征图:")
print(output.squeeze().detach().numpy())

在这个示例中,我们定义了一个 (4 \times 4) 的输入图像和一个 (2 \times 2) 的卷积核。卷积操作的步幅为1,无填充。输出特征图的尺寸为 (3 \times 3)。

多通道卷积

# 定义输入图像(3通道)和卷积核(3输入通道,2输出通道)
input_image = torch.randn(1, 3, 4, 4)  # 批量大小1,3通道,4x4图像
conv_kernel = torch.randn(2, 3, 2, 2)  # 2个卷积核,每个核对应3输入通道,大小2x2

# 定义卷积层
conv_layer = nn.Conv2d(in_channels=3, out_channels=2, kernel_size=2, stride=1, padding=0, bias=False)
conv_layer.weight.data = conv_kernel

# 执行卷积操作
output = conv_layer(input_image)

# 打印输出特征图的尺寸
print("输出特征图尺寸:", output.shape)  # 输出: torch.Size([1, 2, 3, 3])

在这个示例中,我们定义了一个 3 × 4 × 4 3 \times 4 \times 4 3×4×4 的输入图像(3个通道)和两个卷积核,每个卷积核对应3个输入通道。卷积操作的步幅为1,无填充。输出特征图的尺寸为 1 × 2 × 3 × 3 1 \times 2 \times 3 \times 3 1×2×3×3,表示批量大小为1,2个输出通道,每个特征图尺寸为 3 × 3 3 \times 3 3×3

通过这些示例,你可以看到如何在PyTorch中实现图像卷积操作,并理解卷积核、填充和步幅等概念对卷积结果的影响。图像卷积能够提取图像的局部特征,是构建卷积神经网络的基础。

3 填充和步幅

卷积操作中的填充(Padding)和步幅(Stride)是控制输出特征图尺寸的重要参数。它们在卷积神经网络(CNN)中被广泛使用,用于调整特征图的大小和采样密度。

3.1 填充(Padding)

填充是指在输入图像的边缘添加额外的像素(通常是零),以控制输出特征图的尺寸。常见的填充方式有:

  • VALID填充:不添加任何填充,输出特征图的尺寸小于或等于输入图像尺寸。
  • SAME填充:添加填充以确保输出特征图的尺寸与输入图像尺寸相同(当步幅为1时)。

填充的计算公式
假设输入图像尺寸为 H × W H \times W H×W,卷积核尺寸为 K × K K \times K K×K,填充大小为 P P P,步幅为 S S S,则输出特征图的尺寸为:

H out = ⌊ H + 2 P − K S ⌋ + 1 H_{\text{out}} = \left\lfloor \frac{H + 2P - K}{S} \right\rfloor + 1 Hout=SH+2PK+1

W out = ⌊ W + 2 P − K S ⌋ + 1 W_{\text{out}} = \left\lfloor \frac{W + 2P - K}{S} \right\rfloor + 1 Wout=SW+2PK+1

示例代码

import torch
import torch.nn as nn

# 定义输入图像
input_image = torch.randn(1, 1, 4, 4)  # 批量大小1,1通道,4x4图像

# 定义卷积层(VALID填充)
conv_layer_valid = nn.Conv2d(in_channels=1, out_channels=1, kernel_size=3, stride=1, padding=0, bias=False)
output_valid = conv_layer_valid(input_image)
print("VALID填充输出特征图尺寸:", output_valid.shape)  # 输出: torch.Size([1, 1, 2, 2])

# 定义卷积层(SAME填充)
conv_layer_same = nn.Conv2d(in_channels=1, out_channels=1, kernel_size=3, stride=1, padding=1, bias=False)
output_same = conv_layer_same(input_image)
print("SAME填充输出特征图尺寸:", output_same.shape)  # 输出: torch.Size([1, 1, 4, 4])
3.2 步幅(Stride)

步幅是指卷积核在输入图像上滑动的步长。较大的步幅会减少输出特征图的尺寸,同时降低计算量。

示例代码

# 定义卷积层(步幅为2)
conv_layer_stride = nn.Conv2d(in_channels=1, out_channels=1, kernel_size=2, stride=2, padding=0, bias=False)
output_stride = conv_layer_stride(input_image)
print("步幅为2的输出特征图尺寸:", output_stride.shape)  # 输出: torch.Size([1, 1, 2, 2])
.3.3 多通道卷积中的填充和步幅

在多通道卷积中,填充和步幅的设置对每个通道的卷积操作都生效。以下示例展示了在多通道卷积中如何设置填充和步幅。

# 定义输入图像(3通道)
input_image = torch.randn(1, 3, 4, 4)  # 批量大小1,3通道,4x4图像

# 定义卷积层(多通道,SAME填充,步幅为2)
conv_layer = nn.Conv2d(in_channels=3, out_channels=2, kernel_size=3, stride=2, padding=1, bias=False)
output = conv_layer(input_image)
print("多通道卷积输出特征图尺寸:", output.shape)  # 输出: torch.Size([1, 2, 2, 2])

在这个示例中,我们对一个3通道的输入图像应用卷积操作,设置填充大小为1,步幅为2。输出特征图的尺寸为 (1 \times 2 \times 2 \times 2),表示批量大小为1,2个输出通道,每个特征图尺寸为 (2 \times 2)。

通过合理设置填充和步幅,可以有效控制卷积神经网络中特征图的尺寸和采样密度,从而优化网络的性能和计算效率。

4 多输入多输出通道

在卷积神经网络(CNN)中,多输入多输出通道的设置使得模型能够处理彩色图像并提取更丰富的特征。每个输入通道对应一种颜色通道(如RGB图像的红、绿、蓝通道),而多个输出通道则允许模型学习不同的特征。

4.1 多输入通道

对于彩色图像(如RGB格式),有3个输入通道,每个通道对应一种颜色。卷积核需要对每个输入通道进行卷积操作,然后将结果相加得到一个输出通道。

示例代码

import torch
import torch.nn as nn

# 定义输入图像(3通道)
input_image = torch.randn(1, 3, 4, 4)  # 批量大小1,3通道,4x4图像

# 定义卷积层(3输入通道,1输出通道)
conv_layer = nn.Conv2d(in_channels=3, out_channels=1, kernel_size=2, stride=1, padding=0, bias=False)
output = conv_layer(input_image)
print("多输入通道卷积输出特征图尺寸:", output.shape)  # 输出: torch.Size([1, 1, 3, 3])

在这个示例中,我们定义了一个 3 × 4 × 4 3 \times 4 \times 4 3×4×4 的输入图像(3个通道)和一个卷积核,该卷积核对应3个输入通道。卷积操作分别对每个输入通道进行,结果相加得到一个输出通道。输出特征图的尺寸为 1 × 1 × 3 × 3 1 \times 1 \times 3 \times 3 1×1×3×3,表示批量大小为1,1个输出通道,每个特征图尺寸为 3 × 3 3 \times 3 3×3

4.2 多输出通道

使用多个卷积核可以生成多个输出通道,每个卷积核提取不同的特征。多个输出通道能够捕捉输入数据的多种特征,使模型更强大。

示例代码

# 定义卷积层(3输入通道,2输出通道)
conv_layer = nn.Conv2d(in_channels=3, out_channels=2, kernel_size=2, stride=1, padding=0, bias=False)
output = conv_layer(input_image)
print("多输出通道卷积输出特征图尺寸:", output.shape)  # 输出: torch.Size([1, 2, 3, 3])

在这个示例中,我们使用多个卷积核来生成多个输出通道。每个卷积核提取不同的特征,输出特征图的尺寸为 1 × 2 × 3 × 3 1 \times 2 \times 3 \times 3 1×2×3×3,表示批量大小为1,2个输出通道,每个特征图尺寸为 3 × 3 3 \times 3 3×3

4.3 多输入多输出通道的卷积操作

多输入多输出通道的卷积操作结合了多输入和多输出通道的特性,允许模型处理彩色图像并提取丰富的特征。

示例代码

# 定义输入图像(3通道)
input_image = torch.randn(1, 3, 4, 4)  # 批量大小1,3通道,4x4图像

# 定义卷积层(3输入通道,2输出通道)
conv_layer = nn.Conv2d(in_channels=3, out_channels=2, kernel_size=2, stride=1, padding=0, bias=False)
output = conv_layer(input_image)
print("多输入多输出通道卷积输出特征图尺寸:", output.shape)  # 输出: torch.Size([1, 2, 3, 3])

在这个示例中,我们定义了一个 3 × 4 × 4 3 \times 4 \times 4 3×4×4 的输入图像(3个通道)和两个卷积核,每个卷积核对应3个输入通道。卷积操作分别对每个输入通道进行,结果相加得到一个输出通道。两个卷积核生成两个输出通道,输出特征图的尺寸为 1 × 2 × 3 × 3 1 \times 2 \times 3 \times 3 1×2×3×3,表示批量大小为1,2个输出通道,每个特征图尺寸为 3 × 3 3 \times 3 3×3

通过多输入多输出通道的卷积操作,模型能够处理彩色图像并提取丰富的特征,为构建更深层次的卷积神经网络奠定了基础。

5 汇聚层

汇聚层(Pooling Layer)是卷积神经网络(CNN)中的一个重要组件,用于降低特征图的空间尺寸,减少计算量,并赋予模型一定的平移不变性。常见的汇聚方式有最大汇聚(Max Pooling)和平均汇聚(Average Pooling)。

5.1 最大汇聚(Max Pooling)

最大汇聚通过提取特征图中每个局部区域的最大值来生成输出特征图。这种方法可以保留图像中的显著特征,同时减少计算量。

数学原理
假设输入特征图 X ∈ R H × W X \in \mathbb{R}^{H \times W} XRH×W,汇聚核大小为 K × K K \times K K×K,步幅为 S S S,则输出特征图 Y ∈ R H out × W out Y \in \mathbb{R}^{H_{\text{out}} \times W_{\text{out}}} YRHout×Wout,其中:

H out = ⌊ H − K S ⌋ + 1 H_{\text{out}} = \left\lfloor \frac{H - K}{S} \right\rfloor + 1 Hout=SHK+1

W out = ⌊ W − K S ⌋ + 1 W_{\text{out}} = \left\lfloor \frac{W - K}{S} \right\rfloor + 1 Wout=SWK+1

示例代码

import torch
import torch.nn as nn

# 定义输入特征图
input_tensor = torch.randn(1, 3, 8, 8)  # 批量大小1,3通道,8x8特征图

# 定义最大汇聚层
max_pool_layer = nn.MaxPool2d(kernel_size=2, stride=2, padding=0)
output_max_pool = max_pool_layer(input_tensor)
print("最大汇聚输出特征图尺寸:", output_max_pool.shape)  # 输出: torch.Size([1, 3, 4, 4])

在这个示例中,我们定义了一个 3 × 8 × 8 3 \times 8 \times 8 3×8×8 的输入特征图,并使用最大汇聚层进行处理。汇聚核大小为 2 × 2 2 \times 2 2×2,步幅为 2 2 2,无填充。输出特征图的尺寸为 3 × 4 × 4 3 \times 4 \times 4 3×4×4

5.2 平均汇聚(Average Pooling)

平均汇聚通过计算特征图中每个局部区域的平均值来生成输出特征图。这种方法可以平滑特征信息,减少噪声。

示例代码

# 定义平均汇聚层
avg_pool_layer = nn.AvgPool2d(kernel_size=2, stride=2, padding=0)
output_avg_pool = avg_pool_layer(input_tensor)
print("平均汇聚输出特征图尺寸:", output_avg_pool.shape)  # 输出: torch.Size([1, 3, 4, 4])

在这个示例中,我们同样定义了一个 (3 \times 8 \times 8) 的输入特征图,并使用平均汇聚层进行处理。汇聚核大小为 (2 \times 2),步幅为 (2),无填充。输出特征图的尺寸为 (3 \times 4 \times 4)。

5.3 汇聚层的作用
  • 降低计算量:通过减少特征图的空间尺寸,降低后续层的计算量。
  • 平移不变性:汇聚操作使得模型对输入数据的小平移具有不变性,提高模型的鲁棒性。
  • 特征提取:汇聚操作保留了特征图中的重要信息,去除冗余信息,有助于提取更高级的特征。

汇聚层通过最大汇聚或平均汇聚的方式,降低特征图的空间尺寸,减少计算量并赋予模型一定的平移不变性。最大汇聚适用于保留显著特征,平均汇聚适用于平滑特征信息。在卷积神经网络中,汇聚层通常紧跟卷积层之后,以优化特征提取过程。

6 卷积神经网络(LeNet)

LeNet 是最早的卷积神经网络之一,由 Yann LeCun 于 1998 年提出,主要用于手写数字识别任务(如 MNIST 数据集)。LeNet 的提出为后续卷积神经网络的发展奠定了基础。

6.1 LeNet 的架构

LeNet 的典型架构包括以下几个部分:

  1. 卷积层:用于提取图像的局部特征。
  2. 激活函数:引入非线性,使得模型能够学习复杂的特征。
  3. 汇聚层:降低特征图的空间尺寸,减少计算量。
  4. 全连接层:对提取的特征进行分类。

具体来说,LeNet 的架构如下:

  • 卷积层 1:输入为 32 × 32 32 \times 32 32×32 的灰度图像,输出 6 个 28 × 28 28 \times 28 28×28 的特征图。
  • 激活函数:使用 Tanh 激活函数。
  • 汇聚层 1:使用平均汇聚,将特征图尺寸减小到 14 × 14 14 \times 14 14×14
  • 卷积层 2:输出 16 个 10 × 10 10 \times 10 10×10 的特征图。
  • 激活函数:使用 Tanh 激活函数。
  • 汇聚层 2:使用平均汇聚,将特征图尺寸减小到 5 × 5 5 \times 5 5×5
  • 全连接层 1:将 16 个 5 × 5 5 \times 5 5×5 的特征图展平为 120 维向量。
  • 全连接层 2:输出 84 维向量。
  • 全连接层 3:输出 10 维向量,表示 10 个类别。
6.2 LeNet 的实现

以下是使用 PyTorch 实现 LeNet 的代码示例:

import torch
import torch.nn as nn
import torch.nn.functional as F

class LeNet(nn.Module):
    def __init__(self):
        super(LeNet, self).__init__()
        self.conv1 = nn.Conv2d(1, 6, kernel_size=5, stride=1, padding=0)
        self.conv2 = nn.Conv2d(6, 16, kernel_size=5, stride=1, padding=0)
        self.fc1 = nn.Linear(16 * 5 * 5, 120)
        self.fc2 = nn.Linear(120, 84)
        self.fc3 = nn.Linear(84, 10)

    def forward(self, x):
        # 卷积层 1
        x = self.conv1(x)
        x = torch.tanh(x)
        x = F.avg_pool2d(x, kernel_size=2, stride=2)

        # 卷积层 2
        x = self.conv2(x)
        x = torch.tanh(x)
        x = F.avg_pool2d(x, kernel_size=2, stride=2)

        # 全连接层
        x = x.view(x.size(0), -1)  # 展平特征图
        x = self.fc1(x)
        x = torch.tanh(x)
        x = self.fc2(x)
        x = torch.tanh(x)
        x = self.fc3(x)
        return x

# 测试 LeNet 模型
if __name__ == "__main__":
    # 创建模型实例
    model = LeNet()

    # 创建输入数据
    input_data = torch.randn(1, 1, 32, 32)  # 批量大小 1,1 通道,32x32 图像

    # 前向传播
    output = model(input_data)

    # 打印输出
    print("输入数据尺寸:", input_data.shape)
    print("输出数据尺寸:", output.shape)
    print("输出数据:", output)
代码解释
  1. 卷积层

    • self.conv1:第一个卷积层,输入通道数为 1,输出通道数为 6,卷积核大小为 5。
    • self.conv2:第二个卷积层,输入通道数为 6,输出通道数为 16,卷积核大小为 5。
  2. 激活函数

    • 使用 torch.tanh 作为激活函数,引入非线性。
  3. 汇聚层

    • 使用 F.avg_pool2d 进行平均汇聚,汇聚核大小为 2,步幅为 2。
  4. 全连接层

    • self.fc1:将 16 个 5 × 5 5 \times 5 5×5 的特征图展平为 120 维向量。
    • self.fc2:将 120 维向量映射到 84 维向量。
    • self.fc3:将 84 维向量映射到 10 维向量,表示 10 个类别。
  5. 前向传播

    • 输入图像经过两个卷积层和两个汇聚层,提取特征后展平,再通过三个全连接层进行分类。

通过这个示例,你可以了解 LeNet 的基本结构和实现方法。LeNet 是卷积神经网络的开山之作,虽然现在已经被更复杂的网络所取代,但它的设计理念和架构仍然具有重要的参考价值。

总的来说,卷积神经网络通过卷积层和汇聚层的组合,能够有效地提取图像中的局部特征并降低计算复杂度。多输入多输出通道的设计使得CNN能够处理复杂的图像数据。LeNet 作为经典的卷积神经网络模型,展示了CNN在图像识别任务中的强大能力。


网站公告

今日签到

点亮在社区的每一天
去签到