手写体数字识别例程——LeNet-5模型

本文介绍了在Caffe中训练LeNet-5模型来识别手写数字的步骤,包括获取MNIST数据集、转换数据格式、配置网络参数、训练网络和分析训练日志。通过训练,模型的预测准确率达到了0.9912。
摘要由CSDN通过智能技术生成
   上一篇博客中介绍了Caffe环境的搭建,本片博客中介绍一下,在caffe中训练的第一个CNN模型LeNet-5。如果存在不正确的地方欢迎指正。
   该例程用的数据集是MNIST,该数据集中包含60000个训练集和10000个测试集。使用的CNN模型是LeNet-5网络模型,该模型包含2个卷积层,两个下采样层和两个全连接层。
   本文第一部分介绍,在Ubuntu 14.04系统下,如何训练LeNet-5模型。第二部分介绍该模型的prototxt文件。第三部分介绍训练日志中的一些信息及一些卷积的输入输出对照关系。

##Caffe中训练LeNet-5的方法

1、获取mnist数据集
在caffe目录下,执行data/mnist/ 目录下的get_mnist.sh

./data/mnist/get_mnist.sh

执行之后,查看该目录可以看到该目录下增加了四个文件:
这里写图片描述
从命名方式上可以看出,四个文件分别为测试集的图片和标签,训练集的图片和标签。

2、把数据转换成leveldb格式
下载的数据集是二进制格式,需要转换为LEVEDB或者LMDB才能被caffe识别,需要进行数据格式转换。

./example/mnist/create_mnist.sh

3、训练网络

./example/mnist/train_lenet.sh

##查看prototxt文件

  • 查看训练的配置文件
vi examples/mnist/lenet_sover.prototxt

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值