首页 > 代码库 > 【caffe-windows】 caffe-master 之 cifar10 超详细

【caffe-windows】 caffe-master 之 cifar10 超详细

本教程尽量详细,大多步骤都有图,如果运行出错,请先对照自己的文件是否和图上的一样,包括标点啊,空格啊,斜杠,反斜杠啊之类的小细节。

本例程是在 win10 64位   caffe-master    vs2013下进行的,并且已经配置GPU版本,若用CPU,则在cifar10_quick_solver.prototxt中更改

# solver mode: CPU or GPU

solver_mode: CPU

 

 

摘要:win10下的caffe例子:cifar10

 

第一步:

官网下载bin格式的cifar10  : http://www.cs.toronto.edu/~kriz/cifar.html

解压存放在 :caffe-master\examples\cifar10\input_folder当中(input_folder文件夹需要自己创建)

 

如图:

 

技术分享

 

 

第二步: 

在caffe-master\examples\cifar10 下创建一个记事本(后缀改为  .bat ),输入以下code

 

convert_cifar_data.exe input_folder output_folders leveldb 

pause 

 

保存为convert_cifar10.bat (后缀改为  .bat)

 

如图:

 

技术分享

 

 

 

然后双击运行,可以看到在 caffe-master\examples\cifar10下会生成一个文件夹,output_folders。里面存放的就是转换好的数据

 

第三步:

计算出均值文件:mean.binaryproto

首先要确保在caffe-master\Build\x64\Release 下有convert_cifar_data.exe ,没有的话请打开Caffe.sln,将compute_image_mean 设置为 启动项目,然后ctrl+F5,等待编译结束,在caffe-master\Build\x64\Release下就有convert_cifar_data.exe了

 

如图:

 

技术分享

 

有了convert_cifar_data.exe,我们要在caffe-master\examples\cifar10下创建一个记事本(后缀改为  .bat ),输入以下code

 

..\..\Build\x64\Release\compute_image_mean.exe-backend=leveldb ../../examples\cifar10\output_folders\cifar10_train_leveldbmean.binaryproto

Pause

 

然后运行此文件,在caffe-master\examples\cifar10 下会出现mean.binaryproto文件

 

第四步:

更改caffe-master\examples\cifar10 下的cifar10_quick_solver.prototxt和 cifar10_quick_train_test.prototxt  主要是改路径

我的cifar10_quick_solver.prototxt改中了:

net:"../../examples/cifar10/cifar10_quick_train_test.prototxt"

snapshot_prefix:"../../examples/cifar10/cifar10_quick"

cifar10_quick_train_test.prototxt中改了:

mean_file:"../../examples/cifar10/mean.binaryproto"

source:"../../examples/cifar10/output_folders/cifar10_train_leveldb"

mean_file:"../../examples/cifar10/mean.binaryproto"

source:"../../examples/cifar10/output_folders/cifar10_test_leveldb"

 

以及 backend: LEVELDB  (有两处喔!)

 

第五步:

更改好两个文件之后就可以运行caffe来训练啦,提供两种方法,一种是编写批处理文件bat,一种是直接在vs2013中运行

先说第一种,在caffe-master\examples\cifar10下创建文本文件,更改后缀为.bat,复制以下代码:

 

..\..\Build\x64\Release\caffe.exe train--solver=../../examples/cifar10/cifar10_quick_solver.prototxt

Pause

 

如图:

 

技术分享

 

保存之后,运行即可。

 

第二种方法是在vs2013里进行的,先将caffe设置为启动项目:

 

如图:

 

技术分享

 

 

再点击属性,找到调试,在命令参数中输入:train --solver=../../examples/cifar10/cifar10_quick_solver.prototxt

 

如图:

 

技术分享

 

点击确定,ctrl+F5 即可

 

 

【caffe-windows】 caffe-master 之 cifar10 超详细