本文延续上篇文章《Caffe安装教程:Ubuntu16.04(CPU)》(http://www.linuxdiyf.com/linux/26229.html),对搭建好的caffe使用MNIST数据集进行测试,可以更好的验证Caffe。
1.下载数据
cd ~/caffe/
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh
可以查看下载的数据:
2.修改配置
因为是CPU运行,所以修改在examples文件下的Mnist下的lenet_solver.prototxt中的solver_mode:CPU
cd ~/caffe/
sudo gedit ./examples/mnist/lenet_solver.prototxt
修改为下图:
solver_mode: CPU
3.训练模型
cd ~/caffe/
./examples/mnist/train_lenet.sh
4.结果展示
5.可能遇到的问题
可能会遇到./create_mnist.sh: 17: ./create_mnist.sh: build/examples/mnist/convert_mnist_data.bin: not found这样的错误,这是因为新版caffe都需要从根目录上执行,这里我们是从mnist目录下来执行的,所以我们需要先执行:
cd ~/caffe/
进入根目录,然后执行命令:
./examples/mnist/train_lenet.sh