Caffe的安装及MNIST数据集训练

Caffe的安装及MNIST数据集训练

准备caffe环境

打开终端执行命令

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
git clone https://github.com/BVLC/caffe
cd Desktop/caffe/

这时候需要将Makefile.config.example拷贝一份成Makefile.config，并修改# CPU := 1,将前面的#去掉，因为我们要用的是CPU

cp Makefile.config.example Makefile.config
make all
make test
make runtest

这里需要修改文件里

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/

不需要删东西，只要把这两行覆盖原来的就好了

最终 make runtest

表示安装成功

需要注意的是，在安装的过程中要确保依赖包安装的全，其次就是Makefile.config一定要修改对

MNIST训练

mnist为caffe自带数据集，但是在训练之前仍然要修改文件

需要修改./examples/mnist/lenet_solver.prototxt最后一句话为

solver_mode:CPU

下载数据集

cd caffe/data/mnist/
./get_mnist.sh

回到caffe目录下

./examples/mnist/create_mnist.sh

训练

​ 由于是CPU，超级慢…

./examples/mnist/train_lenet.sh

对训练好的模型进行测试