2:使用MNIST数据集验证Caffe是否安装成功

Posted 莫欺丶少年穷

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了2:使用MNIST数据集验证Caffe是否安装成功相关的知识,希望对你有一定的参考价值。

使用MNIST数据集验证Caffe是否安装成功

本文延续上篇博文《Caffe安装教程:Ubuntu16.04(CPU) 》,对搭建好的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

以上是关于2:使用MNIST数据集验证Caffe是否安装成功的主要内容,如果未能解决你的问题,请参考以下文章

使用caffe训练mnist数据集 - caffe教程实战

caffe的学习和使用·一」--使用caffe训练自己的数据

CAFFE学习笔记将自己的jpg数据转成lmdb格式

获取minist数据并转换成lmdb

CAFFE安装(10):Mnist测试(可不做)

21天学习caffe