caffe运行过程

来源:互联网 发布:淘宝达人 自媒体 认证 编辑:程序博客网 时间:2024/05/17 20:28

1、先下载好数据,然后将数据转换为caffe需要的数据格式,leveldb格式(tf的格式是tfrecord)。这个一般有已写好的脚本,用户需要做的是修改这样的脚本,改成属于自己的图片路径。(也可以自己制作图片,但是要遵循一定的格式)
2、编写网络结构文件,这个文件后缀是.prototxt,就是编写你的网络有多少层,每一层有多少个特征图,输入,输出…………(参考caffe自带的例程)
3、网络求解文件,这个文件一般取名为solver.prototxt,这个文件的后缀格式也是.prototxt,这个文件主要包含了一些求解网络,梯度下降参数,迭代次数等参数(主要是一些超参数)
4、训练
1)直接训练
命令行:TOOLS=../cafferead/build/tools
$TOOLS/caffe train –solver=gender_solver.prorotxt -gpu all #使用所有的gpu

2)采用funing-tuning训练法
TOOLS=../cafferead/build/tools
$TOOLS/caffe train –solver=gender_solver.prorotxt -weights gender_net.caffemodel #加入-weights

5、训练完毕后,我们就可以得到caffe的训练模型,接着我们的目标就预测,看看结果了。caffe为我们提供了方便调用的python接口函数,这些都在模块pycaffe里面。提供很多接口操作模型。