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

caffe编译完成后还要继续编译convert_mnist_siamese_data,相关的库全部编译后面的使用才不会出现问题

# sudo make clean

# sudo make convert_mnist_data

# sudo make convert_mnist_siamese_data

# sudo make all

# sudo make install

1.下载数据

cd ~/caffe/
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

1
2
3
1
2
3

可以查看下载的数据： 

2.修改配置

因为是CPU运行，所以修改在examples文件下的Mnist下的lenet_solver.prototxt中的solver_mode:CPU

cd ~/caffe/
sudo gedit ./examples/mnist/lenet_solver.prototxt

1
2
1
2

修改为下图：

solver_mode: CPU

1
1

3.训练模型

cd ~/caffe/
./examples/mnist/train_lenet.sh

1
2
1
2

4.结果展示

5.可能遇到的问题

可能会遇到./create_mnist.sh: 17: ./create_mnist.sh: build/examples/mnist/convert_mnist_data.bin: not found这样的错误，这是因为新版caffe都需要从根目录上执行，这里我们是从mnist目录下来执行的，所以我们需要先执行：

cd ~/caffe/

1
1

进入根目录，然后执行命令：

./examples/mnist/train_lenet.sh