caffe学习笔记——用lenet网络及mnist数据集测试caffe

本文通过一个一个最简单的网络结构lenet来对刚才安装的caffe进行测试:

1、将下载的mnist数据集解压到D:\caffe\caffe-master\caffe-master\data\mnist:

2、转换数据格式

       转换成caffe需要的数据格式,此处转换为 LMDB(cifar是转换成 LEVELDB,可对比参考如何实现的),转换格式需要用到caffe里的项目convert_mnist_data ,这个convert_mnist_data.exe程序是放在caffe-master\Build\x64\Release下,如果是用 debug生成的解决方案,则是在caffe-master\Build\x64\Debug下,这个在写 bat文件的时候要注意更改相应的路径。 (PS:想转换成LEVELDB格式的话,只需要在.bat文件中把 mnist_train_lmdb改成mnist_train_leveldb ,有两处喔! )

(1)在caffe的根目录caffe_master下新建一个脚本文件create_mnist.bat


(2)在脚本文件中写入如下内容:


其实第一个路径就是 exe文件所在路径, 后面两个就是数据所在路径,最后一个是输出文件路径

(3)保存脚本后,双击运行:


则在D:\caffe\caffe-master\caffe-master\examples\mnist目录下生成对应文件夹:




3、修改D:\caffe\caffe-master\caffe-master\examples\mnist\lenet_solver.prototxt,将最后一行改为solver_mode:CPU,修改D:\caffe\caffe-master\caffe-master\examples\mnist\lenet_train_test.prototxt,如下所示,左面为原始的,右面为修改后的。


4、在caffe根目录下,新建train_mnist.bat,然后输入如下的脚本:


保存后双击运行,就会开始训练,训练完毕后会得到相应的准确率(0.991)和损失率。


期间遇到的错误:

1.Debug 和 Release ,这个是在VS2013中选择的,在调用例如 convert_mnist_data.exe 、compute_image_mean.exe等这些 应用程序时,要选择路径,一定要看清楚你生成的是在Debug文件夹(caffe-master\Build\x64\Debug)里还是在Release(caffe-master\Build\x64\Release)里,如果你生成的是在Debug里,而调用的时候是用路径:. Build\x64\Release 的话,就会出现如下错误:


可以在VS2013中分别 选择 Debug和Release 进行生成解决方案,问题就不会出现了。

2、转换数据时候要确定转换的是 LMDB还是LEVELDB,这个是在一开始转换的时候设置的,一旦设置,在后面的train_test.prototxt文件中一定要对应的改过来。


猜你喜欢

转载自blog.csdn.net/wanty_chen/article/details/80199543