首页 > 代码库 > caffe-win10-cifar10

caffe-win10-cifar10

因为是在win10下安装的GPU版caffe,所以不能直接运行linux里的shell脚本。但是win10自带bash,可以运行.sh文件,网上也有直接下Cygwin和git的。我是下载好git后才知道win10自带有bash。。

win10自带bash:

技术分享

git:

技术分享

 转入正题,运行cifar10实例。

在运行这个实例过程中,给我体会最深的是路径问题。我的caffe路径是:

E:\caffe-master\caffe-master\Build\x64\Release

E:\caffe-master\caffe-master\examples\cifar10

下图所示路径即E:\caffe-master\caffe-master就是caffe的根路径。

技术分享

 

1.下载数据:

# sudo sh data/cifar10/get_cifar10.sh

 以上命令在caffe根路径执行即可。

运行成功后,会在 data/cifar10/文件夹下生成一堆bin文件:原文件中只有get_cifar10这一shell文件。

技术分享

可见生成的5个训练集和一个测试集。

 

2.数据转换

bin文件为二进制文件,而caffe只许leveldb和lmdb两种数据格式。这里选择转为leveldb格式。

在/examples/cifar10文件夹中有一个 convert_cifar_data.cpp文件,经过编译之后会生成convert_cifar_data.exe可执行文件。注意这个文件的位置在:

技术分享

可能有人文件位置不在这个地方,两种原因:第一种是最开始下载安装caffe的时候是在https://github.com/happynear/caffe-windows。而我在https://github.com/BVLC/caffe/tree/windows,两者路径有些不同,第一种而言convert_cifar_data.exe在bin文件下,而我的路径如上图。

新建一个名为input的空文件夹,将之前生成的6个bin(二进制)文件拷贝到input文件夹中,再建一个空文件夹output用来存放生成的leveldb文件。在当前目录下(dara/cifar10/)建立一个 bat 文件,实现binary 到 leveldb 数据集格式转换。(要建立一个output空文件夹,否则会提示could not found files)批处理代码如下:

convert_cifar_data.exe input output leveldb
pause

注意,以上命令中包含convert_cifar_data.exe、input、output,所以执行以上命令时路径下必须同时包括convert_cifar_data.exe、input、output文件夹,也就是说当前路径是\Build\x64\Release则可以直接执行并且会在这个目录下的output中生成leveldb文件。如下图:

技术分享  

好,已经生产levelde文件在output文件夹中:

技术分享

 

3.求数据图像的均值,获取mean.binaryproto 

这一步需要一个可执行文件:comput_image_mean.exe,他在路径和生成leveldb数据的convert_cifar_data.exe文件在同一个路径下。

技术分享

在当前路径\Build\x64\Release下新建一个bat文件,然后输入以下命令:

compute_image_mean.exe --backend=leveldb output/cifar10_train_leveldb mean.binaryproto
pause

注意一定强调后端是leveldb格式!否则默认lmdb会出错。以上命令生成训练集的均值。同理,以上命令是在\Build\x64\Release下执行的,所以生成的mean.biaryproto也在这个路径下;

技术分享

好了,到目前为止,我们有了leveldb数据、均值文件,下一布就可以开始训练了。

 

4.训练cifar10

首先,在examples/cifar10/文件夹中,找到cifar10_quick_trian_test.prototxt。 修改里面的mean_file和source的路径。backend一律改为LEVELDB。 注意是大写!leveldb不可以!!!,
这个prototxt文件中共有6处要改,分别是train和test的mean-file、source、backend:

技术分享

注意在examples/cifar10/中有几种不同的训练模式(quick和full等),注意一一对应。我们利用的是cifar10_quick_trian_test.prototxt网络结构,所以在solver中选择cifar10_quick_solver.prototxt

例如你选择的网络是cifar10_full_train_test,那么你的slver应为cifar10_full_solver:

技术分享

到此一切就绪可以开始了:

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

注意mean_file、sourse等都是在caffe根目录下计算路径的,所以选择在caffe根目录下建立mytrain-cifar10.bat文件,写入以上命令:以上命令都以根目录为标准给出了caffe.exe和cifar10_quick_solver.prototxtde的相对路径:

技术分享

执行:

技术分享

 

 技术分享

 

参考:

http://blog.csdn.net/hyqsong/article/details/51713444    

http://blog.csdn.net/muyouhang/article/details/50688038

caffe-win10-cifar10