如何在新版本caffe里添加新的一层

1、首先写好你添加层的cpp和cu文件,另外在cpp文件的最后写上REGISTER_LAYER_CLASS(LAYERTYPENAME);其中LAYERTYPENAME是你定义的layer的类名,如MyInputLayer,那么就是REGISTER_LAYER_CLASS(MyInput);,注意不是MyInputLayer。

2、然后在相关的头文件里添加层类的声明

3、在proto文件里定义层的类型和message

4、在upgrade_proto.cpp里const char* UpgradeV1LayerType(const V1LayerParameter_LayerType type) {
  switch (type) {}里添加相关代码case...return

5、写prototxt网络模型文件的时候要注意type的格式

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Caffe是一个深度学习框架,它依赖于protobuf(Protocol Buffers)。目前,Caffe可以使用protobuf 3.6.0来编译。因此,如果你想将Caffe与TensorFlow一起使用,你需要调整依赖项版本,确保使用的是protobuf-3.6.0版本来编译Caffe。 关于protobuf的版本,你提到了一个问题与相关的引用内容。你提到Caffe装了3.3.0版本,但你需要的是3.6.0版本。你通过pip的list命令查看了已安装的包,发现安装了protobuf 3.6.0。为了满足你的需求,你打算卸载protobuf 3.6.0。 综上所述,Caffe与protobuf的版本相关。Caffe可以使用protobuf 3.6.0来编译,并且你可以通过卸载安装了的protobuf 3.6.***<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* [Caffe,protobuf 编译安装](https://blog.csdn.net/Tosonw/article/details/91043145)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *3* [【深度学习】Caffe中的Protobuf版本问题](https://blog.csdn.net/phdsky/article/details/80994090)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值