图像分类1-LeNet模型结构及其Pytorch实现

一、LeNet 概况

• 1990年 Yann LeCun 提出, 在1998 年改进 LeNet-5
• 目标是识别图像中的手写数字, 商用: 自动取款机
• Paper: GradientBased Learning Applied to Document Recognition

MNIST 数据集

• 50000个训练数据
• 10000个测试数据
• 图像大小 28×28、10 类

二、LeNet 网络结构

三、LeNet 的 Pytorch 实现

### 搭建LeNet-5模型
import torch.nn as nn
from torchsummary import summary
class MyLeNet(nn.Module):
    # 初始化网络结构
    def __init__(self):
        super(MyLeNet