caffe之利用mnist数据集训练好的lenet_iter_10000.caffemodel模型测试一张自己的手写体数字

1.安装一些基本依赖项:
$ 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 libopenblas-dev liblapack-dev libatlas-base-dev
$ sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
2.下载caffe:
$ git clone https://github.com/BVLC/caffe.git
3.然后编译caffe:

$ cd ~/caffe
$ sudo cp Makefile.config.example Makefile.config
$ make all -j8


4. sudo vi Makefile.config
将Makefile.config中的USE_CUDNN行前的#去掉:
修改opencv的版本

adding to the Makefile.config with

LIBRARIES += glog gflags protobuf leveldb snappy \ lmdb boost_system hdf5_hl hdf5 m \ opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs


5. 保存后重新编译:

$ sudo make clean
$ sudo make all -j8
$ sudo make test  
$ sudo make runtest

6.测试
从网上下载数据
./data/mnist/get_mnist.sh
生成数据,这个脚本的作用是将数据转换成lmdb格式或leveldb格式
./examples/mnist/create_mnist.sh
训练数据

./examples/mnist/train_lenet.sh  

测试

./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

7. 如需测试自己写的字体可参考:

caffe之利用mnist数据集训练好的lenet_iter_10000.caffemodel模型测试一张自己的手写体数字http://blog.csdn.net/xunan003/article/details/73126425

猜你喜欢

转载自blog.csdn.net/qq_28424679/article/details/78246586
今日推荐