温馨提示:在代码森林上机训练之前先使用ubuntu登录系统,密码通过右侧上机界面获取,登录成功以后切换到root用户下,密码和ubuntu的密码是一样的,使用命令:
su root
caffe自带的例子mnist。
mnist是一个手写数字库,由DL大牛Yan LeCun进行维护。mnist最初用于支票上的手写数字识别, 现在成了DL的入门练习库。征对mnist识别的专门模型是Lenet,算是最早的cnn模型了。
mnist数据训练样本为60000张,测试样本为10000张,每个样本为28*28大小的黑白图片,手写数字为0-9,因此分为10类。
官方教程如下:
首先cd到caffe的目录,使用命令:
cd /opt/caffe
切换到caffe的目录以后,可以使用ls命令查看是否有data、examples目录
1、首先下载mnist数据
sh data/mnist/get_mnist.sh
等待命令执行成功以后,在data/mnist/目录下有四个文件:
train-images-idx3-ubyte: 训练集样本 (9912422 bytes)
train-labels-idx1-ubyte: 训练集对应标注 (28881 bytes)
t10k-images-idx3-ubyte: 测试集图片 (1648877 bytes)
t10k-labels-idx1-ubyte: 测试集对应标注 (4542 bytes)
2、执行训练命令LMDB数据,转换成功后,会在 examples/mnist/目录下,生成两个文件夹,分别是mnist_train_lmdb和mnist_test_lmdb,里面存放的data.mdb和lock.mdb,就是我们需要的运行数据。
sh examples/mnist/create_mnist.sh
(如果想运行leveldb数据,请运行 examples/siamese/ 文件夹下面的程序)
需要的配置文件有两个,一个是lenet_solver.prototxt,另一个是train_lenet.prototxt.
首先打开lenet_solver_prototxt (ps:如果无vim使用apt install vim安装)
vim examples/mnist/lenet_solver.prototxt
根据需要,在max_iter处设置最大迭代次数,以及决定最后一行solver_mode,是否要改成CPU
保存退出后,就可以运行这个例子了
time sh examples/mnist/train_lenet.sh