Caffe初试

Posted 2020-09-02 WHLOOK

1.基本概念

　　Caffe是一个比较流行的神经网络框架，它支持C++、Python等语言，容易上手，但是代码貌似不怎么好读，等有空我。。。;)

2.Windows10下的部署

　　我把我Windows下的编译完的bin上传了，如果opencv+cuda+cudnn配置好了的话应该可以直接用，替换掉caffe-windows-master/bin就行

链接: https://pan.baidu.com/s/1pLAW2Yf 密码: 1234 

　　这里有官方编译完成的文件可供下载：https://github.com/BVLC/caffe/tree/windows　

　-----分割------

　　环境：Win10+VS2013+OpenCV+CUDA7.5+Cudnn+Caffe

　　我之前已经配置好了OpenCV2了，其他的差不多从新开始。

　　Caffe现在官方支持Windows了，不过当时配置有点乱，然后用了一个大神整理好的Caffe包：

链接: https://pan.baidu.com/s/1hsqRQEK 密码: 1234

　　以及下面这个依赖包：

链接: https://pan.baidu.com/s/1ge4QLKB 密码: 1234

　　

　　01.在使用Caffe之前要先安装CUDA7.5和Cudnn，可以在官网下载（如果你打算使用CPU的话就算了）

　　02.将Caffe解压，然后把那个依赖包解压，把里面的bin，include，lib放在caffe-windows-master中的3rdparty文件夹中（有替换就替换），将3rdparty/bin加入到环境变量的path里

　　03.运行/src/caffe/proto/extract_proto.bat

　　04.运行buildVS2013文件夹下的MainBuilder.sln（不使用GPU的话用另外一个文件夹里的）

　　05.改变VS到Release和X64模式

　　06.编译，里面还有compute_image_mean,convert_imageset等项目也要编译，编译要好久的，然后编译成功的话会在bin文件夹里生成相应的exe文件（把这个bin加到环境变量path后可以在CMD直接执行这些exe）

　　07.最后编译完成后，基本上是完成了Caffe的部署，至于详细的部署等，可以参见caffe-windows-master目录下的README.md，虽然是英文，但是比较好懂，说的也比较详细。

3.使用Caffe

　　关于Caffe的详细教程网上比较多，比如该系列就不错：

http://www.cnblogs.com/denny402/tag/caffe/

　　然后大概说一下如何使用（以finetuning为例）：

　　01.在一个Caffe项目中，有两个.prototxt文件比较重要，一个是定义网络结构的，一个是xxx_solver.txt（用来定义训练的相关参数，比如学习率等），然后finetuning的话，还需要一个.caffemodel的文件，就是别人训练好的model

　　02.要进行finetuning的话，网络结构，就是那两个.prototxt要和使用的.caffemodel对应的网络一致，可以进行少许修改，比如输出类别数目，学习参数等。

　　03.然后准备好自己的训练集和测试集，如果使用的是自己的样本，想省事的话就不用把图片换成leveldb或lmdb了，caffe是支持直接用图片的，不过训练集和测试集要各有一个图片列表.txt：格式如下：

位置/../xx1.jpg 0
位置/../xx2.jpg 0
位置/../xx3.jpg 0
位置/../xx4.jpg 1
位置/../xx5.jpg 1
位置/../xx6.jpg 1
.....................

　　     后面的数字是该样本所对应的类别，要从0开始。

　　04.然后 打开cmd，定位到caffe-windows-master目录下执行caffe开始训练

caffe train -solver=examples/../xx_solver.prototxt -weights=examples/../xx.caffemodel

4.其他

　　在finetuning中，如果要指定某个层的权值参数不改变，只需要将它的lr_mult的值设置为0。

　　如果不想使用某个层原来的参数，则将它的名字改变就好。