一:caffe安装
前期准备:
1:Windows版本的caffe,官方链接 http://github.com/microsoft/caffe
2:Visual Studio 2013
安装过程:
1:将下载的caffe-master解压,并将其windows文件夹下的CommonSettings.props.example重命名为CommonSettings.props,如图所示
修改后
2:点击打开
CommonSettings.props,进行CPU下的配置修改,如图所示
修改后
3:将该windows目录下的Caffe.sln用VS2013打开,生成解决方案
这时候弹出来一个框,表示正在还原Nuget程序包,等待一段时间
成功后,会在桌面上出现NugetPackages文件夹,如图
出现了如下错误,这时候需要对libcaffe重新生成一次
出现如下问题后点击该错误,然后保存弹出来的文件,重新生成libcaffe,提示成功
最后重新生成解决方案,提示全部成功
4:配置属性(注意我这里并没有将Debug模式改为Release模式)
5:运行caffe.cpp,按ctrl+F5进行编译,成功后界面如下
二:mnist测试
1: 去官网 http://yann.lecun.com/exdb/mnist下载mnist数据集,然后解压到caffe_master\data\mnist文件夹下,如图
2: 在caffe-master根目录下,新建creat_mnist文本文档,里面写入如下的脚本(注意路径要与自己的保持一致)
.\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\train-images.idx3-ubyte .\data\mnist\train-labels.idx1-ubyte .\examples\mnist\mnist_train_lmdb
echo.
.\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\t10k-images.idx3-ubyte .\data\mnist\t10k-labels.idx1-ubyte .\examples\mnist\mnist_test_lmdb
pause
之后将其另存为creat_mnist.bat文件,特别注意先将扩展名显示出来才能修改
win7方法如下:打开 组织->文件夹和搜索选项->查看,然后将 隐藏已知的扩展名前面的对勾去掉
win10方法如下:直接将 查看->文件扩展名前面的对勾去掉
双击该脚本运行,即可在\caffe-master\examples\mnist下面生成相应的lmdb数据文件
3: 修改\caffe\examples\mnist\lenet_solver.prototxt,将最后一行改为solver_mode:
CPU,
修改\caffe\examples\mnist\lenet_train_test.prototxt,正确指定训练集和测试集,如图
4:
在caffe-master根目录下,新建train_mnist.bat(方法同上),然后输入如下的脚本
.\Build\x64\Debug\caffe.exe train --solver=.\examples\mnist\lenet_solver.prototxt
pause
最后双击运行该脚本,就会得到如下结果,并最终在\examples\mnist下产生训练出的模型,万里长征终于走完了第一步
三:参考
赵永科. 深度学习:21天实战Caffe[M]. 电子工业出版社, 2016.
https://www.cnblogs.com/cxyxbk/p/5902034.html
https://blog.csdn.net/qq_14845119/article/details/52415090
https://www.cnblogs.com/yixuan-xu/p/5858595.html