1 定义solve.prototxt
主要定义一些训练的参数;
2 定义mode.prototxt
主要定义模型结构
3 训练模型
方法1:
build/tools/caffe train -solver /home/www/solve.prototxt
build/tools/caffe train -solver =/home/www/solve.prototxt
方法2:
建立一个py文件 trainCaffe.py
import caffe
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.SGDSolver(’/home/xxx/data/solver.prototxt’)
solver.solve()
运行
python trainCaffe.py
4自定义caffe网络层
添加新的层AallPassLayer实现,需要同其他的Layer类一样,分成声明和实现两个部分,对应放在.hpp和.cpp文件中,如果有cuda实现,还应有.cu文件。其中.hpp头文件放在/caffe/include/caffe/layers/文件夹下,而 .cpp 和 .cu 放入/caffe/src/caffe/layers下
1)头文件的实现
2)cpp文件的实现
3)修改caffe.proto,利用ProtoBuffer生成caffe.pb.h和caffe.pb.cc文件,编译生成caffe.lib库。