Mnist数据集测试Caffe

1、准备数据

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

2、修改配置
修改该目录下的prototxt扩展名配置文件: 修改./examples/mnist/lenet_solver.prototxt, 定位到最后一行:solver_mode: GPU,将GPU改为CPU。 使用CPU进行测试.

3、运行

sudo sh ./examples/mnist/train_lenet.sh

最终训练完的模型存储为一个二进制的protobuf文件:
./examples/mnist/lenet_iter_10000.caffemodel

这个模型就可以用来直接使用了

参考博客:http://blog.csdn.net/lanseyuansu/article/details/70173606

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值