Ubuntu15.04 运行caffe+mnist的demo

来源:互联网 发布:tk的域名 编辑:程序博客网 时间:2024/06/06 08:31

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更换即可在不同条件下运行


0 0
原创粉丝点击