1、准备数据
cd caffe
sudo sh ./data/mnist/get_mnist.sh
sudo sh ./examples/mnist/create_mnist.sh
2、修改配置
修改该目录下的prototxt扩展名配置文件: 修改./examples/mnist/lenet_solver.prototxt, 定位到最后一行:solver_mode: GPU,将GPU改为CPU。 使用CPU进行测试.
3、运行
sudo sh ./examples/mnist/train_lenet.sh
最终训练完的模型存储为一个二进制的protobuf文件:
./examples/mnist/lenet_iter_10000.caffemodel
这个模型就可以用来直接使用了
参考博客:http://blog.csdn.net/lanseyuansu/article/details/70173606