windowscaffe从0到1

首先要明确运行训练一个caffe模型需要哪几个文件

1、网络文件 train.prototxt文件,要修改训练图片、均值文件的路径,校验图片、均值文件的路径

2、solver.prototxt文件,要修改网络的路径和caffe模型的保存路径,我是存放在model4中,在运行程序之前要建好一个model4的文件夹;如果用的cpu把最后一行改成cpu

net: "/home/ths/MF_3_32_90/solver_cccp12_32/train_val_nofully21.prototxt"
test_iter: 60
test_interval: 1000
base_lr: 0.02
lr_policy: "inv"
gamma: 0.01
power: 0.75
#stepsize: 10000
display: 100
max_iter: 100000
momentum: 0.9
weight_decay: 0.0005
snapshot: 2000
snapshot_prefix: "/home/ths/MF_3_32_90/solver_cccp12_32/model4/train_val_cccp12_32input"
solver_mode: GPU

3、运行文件

如果用GPU:

train_caffenet.sh文件内容如下:要修改caffe的存放路径、solver.prototxt的路径、以及最后生成的log文件的存放路径

如果用CPU:

先编写一个run.txt的文件,写好后将后缀名改成bat,双击运行就可以了。具体示例在教程中

明确了训练一个caffe模型需要什么文件,知道每一步都是干什么之后就可以按照这个教程配caffe的环境了,这个教程写的很详细http://www.cnblogs.com/yixuan-xu/p/5858595.html


猜你喜欢

转载自blog.csdn.net/qq_37254336/article/details/80730477