ubuntu16.04+caffe训练mnist数据集
1. caffe-master文件夹权限修改 下载的caffe源码编译的caffe-master文件夹貌似没有写入权限,输入以下命令修改: sudo chmod -R 777 ~/caffe-master/ 2. 下载mnist数据库 cd ~/caffe sduo ./data/mnist/get_mnist.sh caffe中的./data/mnist/get_mnist.sh 文件实现了下载mnist数据库的功能,文件的内容如下: 执行之后,在./data/mnist文件夹下生成4个文件,分别是测试、训练数据库和测试、训练标签: 3. 二进制数据库文件转换成lmdb数据库格式 sudo ./examples/mnist/create_mnist.sh create_mnist.sh文件用于把数据库转换成lmdb格式,内容如下: 执行之后,在 ./examples/mnist文件夹下生成两个文件夹,一个是 mnist_train_lmdb 和 mnist_test_lmdb,分别存放了训练和测试数据: 4. 训练lenet网络 如果只是用CPU训练的话,需要先在 lenet_solver.prototxt 文件中修改训练模式为CPU,使用gedit打开.prototxt文件: sudo gedit ./examples/mnist/lenet_solver.prototxt