mnist

来源:互联网 发布:淘宝网登录首页 编辑:程序博客网 时间:2024/06/06 03:46

1、./data/mnist/get_mnist.sh 获取数据集

2、./data/mnist/get_mnist.sh 数据集转化

3、CAFFE_ROOT/examples/mnist/lenet_train_test.prototxt网络结构

4、CAFFE_ROOT/examples/mnist/lenet_solver.prototxt网络超参数的设置

5、./examples/mnist/train_lenet.sh 训练出模型

6、调用训练好的模型测试
./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/model/lenet_iter_10000.caffemodel -gpu=0(GPU版本)


原创粉丝点击