caffe运行过程

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里面。提供很多接口操作模型。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值