LeNet入门和Pytorch实现

1. LeNet简介

LeNet是一系列网络的合称,包括LeNet1-LeNet5,是卷积神经网络的开山之作。

文献:LeCun Y, Boser B, Denker J, et al. Handwritten digit recognition with a back-propagation network[J]. Advances in neural information processing systems, 1989, 2.

文献:LeCun Y, Bottou L, Bengio Y, et al. Gradient-based learning applied to document recognition[J]. Proceedings of the IEEE, 1998, 86(11): 2278-2324.

2018 年,LeCun因深度学习的研究贡献,与 Yoshua Bengio、Geoffrey Hinton 共同获得了计算机科学最高荣誉图灵奖。

2. LeNet 网络结构

参考:

【深度学习】LeNet网络架构_lenet架构-CSDN博客

LeNet讲解+代码实现,训练MNIST数据集(Pytorch)_使用lenet训练mnist-CSDN博客

深度学习经典网络解析图像分类篇(一):LeNet-5_分类神经网络lenet-CSDN博客

7层神经网络,包括3个卷积层,2个池化层,1个全连接层,1个输出层。其中所有卷积层的卷积核大小都为5*5,步长=1,池化方法为平均池化,激活函数为 Sigmoid(目前使用的Lenet已改为ReLu)。

LeNet-5 网络结构详细解释看这篇,非常详细,我把它截图贴一下,方便查阅。

深度学习经典网络解析图像分类篇(一):LeNet-5_分类神经网络lenet-CSDN博客

下面都是截图截图截图截图截图

公式为: output_size = (input_size - kernel_size + 1),即 (32 - 5 + 1) = 28。

3. 代码实现

  • model.py:定义LeNet网络模型
  • train.py:加载数据集并训练,计算loss和accuracy,保存训练好的网络参数
  • predict.py:用自己的数据集进行分类测试

未完待续

相关推荐
广州灵眸科技有限公司3 分钟前
瑞芯微(EASY EAI)RV1126B AI算法开发流程
人工智能·算法·机器学习
m0_602857761 小时前
如何提升SQL存储过程逻辑复用_封装通用存储过程函数
jvm·数据库·python
志栋智能2 小时前
运维超自动化:构建弹性IT架构的关键支撑
运维·服务器·网络·人工智能·架构·自动化
傻啦嘿哟2 小时前
如何在 Python 中使用 colorama 库来给输出添加颜色
开发语言·python
薛定猫AI2 小时前
【深度解析】Open Design:用本地优先架构重塑 AI UI 生成工作流
人工智能·ui·架构
forEverPlume2 小时前
mysql如何实现高可用集群架构_基于MHA环境搭建与部署
jvm·数据库·python
嵌入式小企鹅3 小时前
CPU供需趋紧、DeepSeek V4全链适配、小米开源万亿模型
人工智能·学习·开源·嵌入式·小米·算力·昇腾
草莓熊Lotso3 小时前
Vibe Coding 时代:LangChain 与 LangGraph 全链路解析
linux·运维·服务器·数据库·人工智能·mysql·langchain
快乐非自愿4 小时前
RAG夺命10连问,你能抗住第几问?
人工智能·面试·程序员
千匠网络7 小时前
破局出海壁垒,千匠网络新能源汽车跨境出海解决方案
人工智能