用已有的模型来训练自己的数据集(finetune)

1.首先准备自己的数据集,这次的数据集和上篇博客中自己训练的数据集是一样的,按照上次的步骤,生成train_lmdb,val_lmdb和均值文件

2.下载已有的模型这里用的是caffenet模型,下载bvlc_reference_caffenet.caffemodel以及网络train_val.prototxt,solver.prototxt

3.修改train_val.prototxt文件,修改数据集路径以及均值文件路径,batch_size的数值由数据集决定,这里train的batchsize=128,test的batchsize=10

   修改最后的num_output=5,并且把这一层的名字改成fc_new

4.修改solver.prototxt文件,test_iter=10,test_interval=10,base_lr=0.001,stepsize=40,display=20,weight_decay=0.0005,这里用CPU训练

5.finetune,./build/tools/caffe train --solver=examples...solver.prototxt -weights=...bvlc_reference_caffenet.caffemodel

6.迭代50次精度就达到100%。

7.测试:./build/tools/caffe.bin test -model examples..train_val_prototxt -weights=examples/...(finetune之后的模型文件) -iterations=10(因为test batchsize=10,迭代10次就能遍历整个测试集)


  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值