1.获取mnist训练和测试数据
cd data/mnist
sudo sh get_mnist.sh
2.转换数据成lmdb
sudo sh examples/mnist/create_mnist.sh
3.配置
因为不用GPU 所以修改examples/mnist/lenet_solver.prototxt文件,将solver_mode:GPU改为 solver_mode:CPU
4.训练
sudo sh ./examples/mnist/train_lenet.sh
最终得到lenet_iter_10000.caffemodel训练模型文件
5.测试
命令行方式 在examples/mnist下新建test_lenet.sh 内容为:
#!/usr/bin/env sh
build/tools/caffe test --model=examples/mnist/lenet_train_test.prototxt --weights=examples/mnist/lenet_iter_10000.caffemodel -iterations 100
回到caffe根目录执行:
sudo sh examples/mnist/test_lenet.sh
执行完毕图片: