Windows CPU环境下caffe安装及mnist训练过程详解

一:caffe安装

前期准备:
1:Windows版本的caffe,官方链接 http://github.com/microsoft/caffe
2:Visual Studio 2013

安装过程:
1:将下载的caffe-master解压,并将其windows文件夹下的CommonSettings.props.example重命名为CommonSettings.props,如图所示
Windows CPU环境下caffe安装及mnist训练过程详解_第1张图片
修改后

2:点击打开 CommonSettings.props,进行CPU下的配置修改,如图所示
Windows CPU环境下caffe安装及mnist训练过程详解_第2张图片
修改后
Windows CPU环境下caffe安装及mnist训练过程详解_第3张图片

3:将该windows目录下的Caffe.sln用VS2013打开,生成解决方案

Windows CPU环境下caffe安装及mnist训练过程详解_第4张图片
这时候弹出来一个框,表示正在还原Nuget程序包,等待一段时间
Windows CPU环境下caffe安装及mnist训练过程详解_第5张图片
成功后,会在桌面上出现NugetPackages文件夹,如图

出现了如下错误,这时候需要对libcaffe重新生成一次

Windows CPU环境下caffe安装及mnist训练过程详解_第6张图片
Windows CPU环境下caffe安装及mnist训练过程详解_第7张图片
出现如下问题后点击该错误,然后保存弹出来的文件,重新生成libcaffe,提示成功
Windows CPU环境下caffe安装及mnist训练过程详解_第8张图片 Windows CPU环境下caffe安装及mnist训练过程详解_第9张图片
Windows CPU环境下caffe安装及mnist训练过程详解_第10张图片
最后重新生成解决方案,提示全部成功
Windows CPU环境下caffe安装及mnist训练过程详解_第11张图片
4:配置属性(注意我这里并没有将Debug模式改为Release模式)
Windows CPU环境下caffe安装及mnist训练过程详解_第12张图片
5:运行caffe.cpp,按ctrl+F5进行编译,成功后界面如下
Windows CPU环境下caffe安装及mnist训练过程详解_第13张图片
Windows CPU环境下caffe安装及mnist训练过程详解_第14张图片


二:mnist测试
1: 去官网 http://yann.lecun.com/exdb/mnist下载mnist数据集,然后解压到caffe_master\data\mnist文件夹下,如图
Windows CPU环境下caffe安装及mnist训练过程详解_第15张图片
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方法如下:打开 组织->文件夹和搜索选项->查看,然后将 隐藏已知的扩展名前面的对勾去掉
Windows CPU环境下caffe安装及mnist训练过程详解_第16张图片
win10方法如下:直接将 查看->文件扩展名前面的对勾去掉
Windows CPU环境下caffe安装及mnist训练过程详解_第17张图片

双击该脚本运行,即可在\caffe-master\examples\mnist下面生成相应的lmdb数据文件


3: 修改\caffe\examples\mnist\lenet_solver.prototxt,将最后一行改为solver_mode: CPU
    修改\caffe\examples\mnist\lenet_train_test.prototxt,正确指定训练集和测试集,如图
Windows CPU环境下caffe安装及mnist训练过程详解_第18张图片
4: 在caffe-master根目录下,新建train_mnist.bat(方法同上),然后输入如下的脚本
.\Build\x64\Debug\caffe.exe train --solver=.\examples\mnist\lenet_solver.prototxt  
pause

最后双击运行该脚本,就会得到如下结果,并最终在\examples\mnist下产生训练出的模型,万里长征终于走完了第一步

Windows CPU环境下caffe安装及mnist训练过程详解_第19张图片

三:参考

赵永科. 深度学习: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


你可能感兴趣的:(Windows CPU环境下caffe安装及mnist训练过程详解)