Ubuntu15.04 运行caffe+mnist的demo


1.      进入Caffe根目录;
2.      加载MNIST数据库:$ ./data/mnist/get_mnist.sh
3.      转换成Lmdb数据库格式:$ ./examples/mnist/create_mnist.sh 执行完此shell脚本后,会在./examples/mnist下增加两个新目录,mnist_test_lmdb和mnist_train_lmdb
4.      训练模型:$ ./examples/mnist/train_lenet.sh
a、调用LeNet网络
b、调用./examples/mnist/lenet_train_test.prototxt模型;

c、调用./examples/mnist/lenet_solver.prototxt模型;

 配置文件中GLOG_logtostderr=1设置打印所有日志消息;train_net.bin训练脚本

d、执行train_lenet.sh脚本,直接调用./build/tools目录下的caffe执行文件,此执行文件的实现是./tools目录下的caffe.cpp文件;
e、执行此脚本后,会生成几个文件,其中./examples/mnist/lenet_iter_10000.caffemodel则是最终训练生成的model文件;

              note:在lenet_solver.prototxt文件中的solver_mode字段值由的GPU和CPU更换即可在不同条件下运行


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值