LeNet入门和Pytorch实现

1. LeNet简介

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

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

文献:LeCun Y, Bottou L, Bengio Y, et al. Gradient-based learning applied to document recognitionJ. 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:用自己的数据集进行分类测试

未完待续

相关推荐
财税数字化1 分钟前
日韩泰票据与欧美发票混合入账怎么选?开灵科技与海外厂商优劣势对比
大数据·人工智能·科技·ocr·软件选型·票据识别
Axis tech10 分钟前
Xsens&Manus全身数据采集解决方案,用精确数据训练您的具身智能
人工智能·科技·机器人
2601_9556624614 分钟前
1300多种音色怎么挑?不同短视频场景的配音思路
人工智能·音视频
技术卷16 分钟前
Windows 下自定义 MCP 踩坑:stdin 默认继承导致 CPython 启动阻塞
python·mcp·zcode
qq_252941316819 分钟前
多模态屋顶损伤与杂物检测数据集 | 屋顶检测 多模态 红外可见光 建筑巡检 目标检测9232期
人工智能·深度学习·yolo·目标检测·计算机视觉·屋顶
小oo呆19 分钟前
【学习心得】Python多进程
python·操作系统
CodeGL20 分钟前
9、Python基础之IO编程
开发语言·数据库·python
java1234_小锋22 分钟前
Arthas 4 宣布正式拥抱 AI
人工智能
HySpark23 分钟前
ERes2Net声纹识别短音频匹配不稳定:从音频时长到有效音素的排查实践
python·语音识别·声纹识别·vad·eres2net
TomEval26 分钟前
【测AI】第17篇:大模型评测方法论 —— LLM 怎么测?
人工智能·python·aigc·测试用例