Mac下安装caffe并运行MNIST

1,首先安装homebrew包管理工具
$ ruby -e "$(curl-fsSL\http://raw.githubusercontent.com/Homebrew/install/master/install)"
2,安装成功后,利用brew安装依赖包
$ brew install -vd snappy leveldb gflags blogs zip lmdv
$ brew tap homebrew/science
$ brew install protobuf boost wget

此过程中可能有些依赖包会装不上,部分可参考下面的解决办法:

glog安装
$ wget https://google-glog.googlecode.com/files/glog-0.3.3.tar.gz
$ tar zxvf glog-0.3.3.tar.gz
$ cd glog-0.3.3
$ ./configure
$ make -stdlib = libstdc++
$ make install

flags安装 
$ wget  http://github.com/schuhschuh/gflags/archive/master.zip
$ unzip master.zip
$ cd gulags-master
$ mkdir build && cd build
$ export CXXFLAGS="-fPIC" &&cmake ..&& make VERBOSE=1
$ make && make install

lmdb安装
$ git clone https://github.com/LMDB/lmdb
$ cd lmdb/libraries/liblmdb
$ make && make install
3,下载caffe源码
$ git clone https://github.com/bvlc/caffe.git
$ cd caffe/
$ mv Makefile.config.example Makefile.config
$ vi Makefile.config
//修改 CPU_ONLY:=1
4,编译
$ make all 
$ make test
$ make runtest

编译过程中,可能会出现很多warning,不过不用担心,没报错就行。
运行结果

5, 运行MNIST

获取并生成训练集和数据集

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

执行完后在mnist文件夹下会多出两个文件夹:mnist_train_lmdb,mnist_test_lmdb,分别是训练集和测试集。
因为Mac上面没有GPU,所以将./examples/mnist/lenet_solver.prototxt文件中最后一行改为:
solver_mode: CPU
运行minst:

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

成功后,会出现如下信息:

yuanfeideMacBook-Pro:caffe figo$ ./examples/mnist/train_lenet.sh
I1107 16:53:39.949373 2009157632 caffe.cpp:210] Use CPU.
I1107 16:53:39.950474 2009157632 solver.cpp:48] Initializing solver from parameters: 
test_iter: 100
test_interval: 500
base_lr: 0.01
display: 100
max_iter: 10000
lr_policy: "inv"
gamma: 0.0001
power: 0.75
momentum: 0.9
weight_decay: 0.0005
snapshot: 5000
snapshot_prefix: "examples/mnist/lenet"
solver_mode: CPU
net: "examples/mnist/lenet_train_test.prototxt"
train_state {
  level: 0
  stage: ""
}
...//中间太长就省略了。。。
I1107 16:53:39.981427 2009157632 net.cpp:228] label_mnist_1_split does not need backward computation.
I1107 16:53:39.981441 2009157632 net.cpp:228] mnist does not need backward computation.
I1107 16:53:39.981448 2009157632 net.cpp:270] This network produces output accuracy
I1107 16:53:39.981456 2009157632 net.cpp:270] This network produces output loss
I1107 16:53:39.981469 2009157632 net.cpp:283] Network initialization done.
I1107 16:53:39.981591 2009157632 solver.cpp:60] Solver scaffolding done.
I1107 16:53:39.981662 2009157632 caffe.cpp:251] Starting Optimization
I1107 16:53:39.981693 2009157632 solver.cpp:279] Solving LeNet
I1107 16:53:39.981703 2009157632 solver.cpp:280] Learning Rate Policy: inv
I1107 16:53:39.983294 2009157632 solver.cpp:337] Iteration 0, Testing net (#0)
I1107 16:53:42.728090 2009157632 solver.cpp:404]     Test net output #0: accuracy = 0.1544
I1107 16:53:42.728124 2009157632 solver.cpp:404]     Test net output #1: loss = 2.32693 (* 1 = 2.32693 loss)
I1107 16:53:42.773195 2009157632 solver.cpp:228] Iteration 0, loss = 2.33117
I1107 16:53:42.773283 2009157632 solver.cpp:244]     Train net output #0: loss = 2.33117 (* 1 = 2.33117 loss)
I1107 16:53:42.773350 2009157632 sgd_solver.cpp:106] Iteration 0, lr = 0.01
I1107 16:53:46.592530 2009157632 solver.cpp:228] Iteration 100, loss = 0.18131
I1107 16:53:46.592572 2009157632 solver.cpp:244]     Train net output #0: loss = 0.18131 (* 1 = 0.18131 loss)
I1107 16:53:46.592582 2009157632 sgd_solver.cpp:106] Iteration 100, lr = 0.00992565
I1107 16:53:50.332984 2009157632 solver.cpp:228] Iteration 200, loss = 0.125354
I1107 16:53:50.333026 2009157632 solver.cpp:244]     Train net output #0: loss = 0.125354 (* 1 = 0.125354 loss)
I1107 16:53:50.333035 2009157632 sgd_solver.cpp:106] Iteration 200, lr = 0.00985258

生成训练集和测试集的时候出现了如下错误:

Library not loaded: /usr/local/opt/protobuf/lib/libprotobuf.9.dylib
  Referenced from: /Users/figo/caffe/build/examples/mnist/convert_mnist_data.bin
  Reason: image not found

出现此错误主要是因为之前安装了caffe,后来安装tensorflow的时候将protobuf更新到了最新的版本,要解决此问题,只需要将caffe重新编译下就可以了。
重新编译的过程中,又出现了如下错误:

.build_release/src/caffe/proto/caffe.pb.h:17:2: error: This file was generated by an older version of protoc

需要clean一下,再编译就行了

$ make clean
$ make all 
$ make test
$ make runtest
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值