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:用自己的数据集进行分类测试

未完待续

相关推荐
不喝水就会渴2 分钟前
【共创季稿事节】HarmonyOS 7.0 时代的新基建 :DevEco CLI + Claude Code,鸿蒙 AI 开发的黄金搭档
人工智能·华为·harmonyos
星河耀银海4 分钟前
大模型和搜索引擎到底有什么不一样
人工智能·搜索引擎
沪漂阿龙4 分钟前
《LangChain》成本、限流、缓存、降级:AI 应用上线要考虑的问题
人工智能·langchain
一切皆是因缘际会8 分钟前
RLHF奖励坍塌:大模型Reward漂移机理
人工智能·数学建模·ai
阿庆_AI研发工程师12 分钟前
从 OpenAI Codex 源码看生产级 AI Agent Runtime 的工程模式
人工智能
武子康13 分钟前
调查研究-177 Agent / Harness 工具链研究:从会调用工具的 LLM,到可观测、可验证、可交付的智能体系统
人工智能
集芯微电科技有限公司14 分钟前
四通道2A输出集成功率电感降压模块专为紧凑型方案设计
人工智能·单片机·嵌入式硬件·生成对抗网络·计算机外设
beethobe15 分钟前
PythonQt 学习之旅(一):从零构建 C++ 与 Python 的桥梁
c++·python·学习
广州智造15 分钟前
如何在HyperMesh运行Python脚本及查找Python API帮助
python·仿真·cae·hypermesh·optistruct
朱大喜16 分钟前
NumPy 性能优化:内存布局、向量化与原地操作的实战经验
人工智能