首页 > 代码库 > 【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 超详细