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,如图所示
修改后

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.


  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值