本文主要介绍:在进行网络训练和权值微调过程中,需要对solver中的参数进行设置,本文主要介绍solver中的参数设置,以及如何利用自己的数据集,对权值进行微调。
1. solver参数设置
- net: "train_val.prototxt"
- test_iter: 1000
-
-
- test_interval: 1000
-
- base_lr: 0.01
- lr_policy: "step"
-
-
-
-
-
-
-
- gamma: 0.1
- stepsize: 3000
-
-
- display: 20
- max_iter: 12000
- momentum: 0.9
- weight_decay: 0.0005
- snapshot: 5000
- snapshot_prefix: "caffenet_train"
- solver_mode: GPU
注:详细的介绍见 http://blog.csdn.net/langb2014/article/details/50457774
2. 权值微调
2.1 数据转化及网络设置
该过程参考 http://blog.csdn.net/hong__fang/article/details/52425105 中的1到3.2;
2.2 参数微调
(1)创建 finetune.txt ,添加内容
- SET GLOG_logtostderr=1
- ..\caffe-windows-master\bin\caffe.exe train --solver=.\model\solver.prototxt --weights=.\model\bvlc_reference_caffenet.caffemodel
- pause
第二行有三个参数,分别为:
- ..\caffe-windows-master\bin\caffe.exe
- train
- --solver=.\model\solver.prototxt
- --weights=.\model\bvlc_reference_caffenet.caffemodel
注:
<1>以上路径均为相对路径;
<2>路径中不能含有中文。
(2)将finetune.txt 改名为finetune.bat,双击运行。