一、网络结构
LeNet包含多个版本,通常所说的LeNet是指LeNet-5,网络结构如下图:
源代码:https://gitee.com/killf/deep-learning-notes
二、数据集
本文采用mnist
数据集,该数据集包含60000个训练数据和10000个测试数据,图像是大小为(28,28,1)的灰度图像,标签是0-9之间的数字,部分图片如下:
三、训练
训练时,需要将图片进行归一化,即压缩到0-1之间。使用如下代码进行训练:
1 | python train.py |
训练过程如下:
四、测试
使用如下代码进行测试:
1 | python test.py 3 |
测试结果如下:
- 本文作者: killf
- 本文链接: http://www.killf.info/机器学习/LeNet/
- 版权声明: 本博客所有文章除特别声明外,均采用 Apache License 2.0 许可协议。转载请注明出处!