Caffe学习(2):在Ubuntu下运行手写体数字教程

来源:互联网 发布:mimo软件 编辑:程序博客网 时间:2024/06/05 15:29

上一节安装好Ubuntu后,通过运行mnist熟悉Caffe的基本使用。
一共分别以下几个步骤:

一、下载mnist数据集

mnist数据集可以在caffe自带的源码get_mnist.sh脚本下载。命令为:

$sudo cd caffe$sudo cd data/mnist#这里的./get_mnist.sh应该就是执行这个文件夹下脚本的意思,虽然我也没太弄懂他们的写法规则$sudo ./get_mnist.sh

执行后,用$tree的命令可以查看文件夹的数据列表。

二、转换格式

下载的原始数据集为二进制文件,要转换成LEVELDB或LMDB格式才能被识别。一方面是为了统一数据格式,简化数据读取层的实现;另一方面是使用LMDB和LEVELDB格式可以提高磁盘利用率。
同样,可以用caffe自带的源码create_mnist.sh转换格式。命令为:

#在caffe根目录下运行$ sudo ./examples/mnist/create_mnist.sh

执行后,同样可以examples/mnist目录下,用$tree命令查看文件数据列表,可以看到mnist_train_lmdb和mnist_test_lmdb两个目录。

三、训练超参数

用caffe自带的LeNet CNN模型训练mnist数据集,用到的源码是train_lenet.sh脚本。
(1)首先,修改lenet_solver.prototxt文件。用命令:

$ sudo nano lenet_solver.prototxt

进入该文件内容,修改文件中最后一行为:solver_mode:CPU,表示用cpu模式运行程序。
(2)在caffe根目录下输入命令:

$sudo ./examples/mnist/train_lenet.sh

正式运行mnist数据,大概花了十几分钟,最后可以看到Test net output #0:accuracy=0.9904,表示test的准确率为99.04%。
因为电脑运行ubuntu有点慢,具体画面就不贴出来。

阅读全文
0 0
原创粉丝点击