1.获取数据
cd caffe/data/mnist
执行命令:./get_mnist.sh
执行完命令后切换到caffe/data/mnist目录下发现get_mnist.sh文件是原有的,其他四个文件是下载的。如图1所示
2.将数据转化为lmdb格式
3.训练
![]()
![]()
cd caffe/data/mnist
执行命令:./get_mnist.sh
执行完命令后切换到caffe/data/mnist目录下发现get_mnist.sh文件是原有的,其他四个文件是下载的。如图1所示
图1
2.将数据转化为lmdb格式
命令:./examples/mnist/create_mnist.sh
切记必须在caffe的根目录下运行否则报错
3.训练
命令:./examples/mnist/train_lenet.sh
切记也必须在caffe的根目录下运行否则报错
4.结果如下图2所示
图2
用GPU训练的一分钟不到就跑完了,准确率99.02%比网上说的好像低了点。
5.如果你没有gpu那就用cpu跑的话可能20分钟左右
执行以下命令修改CPU还是GPU跑:cd caffe/examples/mnist/
vim lenet_solver.prototxt
然后修改成GPU或者CPU即可,如下图3所示
图3