caffe的使用

15 篇文章 0 订阅
10 篇文章 0 订阅

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库。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值