caffe中加入python layer的方法

谁借莪１个温暖的怀抱￠ 2022-05-16 02:25 197阅读 0赞

小鱼趁这个机会，学学python layer 的编写及实现  
要想自己定义网络层的实现，如果c++代码够强大，可以在caffe底层编写cpp文件，进行caffe的编译  
但是如果c++能力不是特别强，可以尝试使用caffe下的python layer实现  
1、 Makefile.config的

#WITH_PYTHON_LAYER := 1改为WITH_PYTHON_LAYER := 1

Caffe目录重新编译Caffe

make clean
            make all

2、将python路径加入到环境变量中

sudo vim ~/.bashrc
    再最后添加export PYTHONPATH = /your/caffe/python/path

3、自己编写py文件，结构如下

import caffe
    class MyLayer(caffe.Layer):
      def setup(self, bottom, top):
        pass
    
      def reshape(self, bottom, top):
        pass
    
      def forward(self, bottom, top):
        pass
    
      def backward(self, top, propagate_down, bottom):
        pass

可参考[自己编写py文件][py]  
4、因为caffe是将每一个py文件读取为module，如果不把py文件的路径加入到环境变量的话，将会出现`import error：no module namedXXX`问题  
解决办法

sudo vim /usr/local/lib/python2.7/dist-packages/Mymodule.pth
    创建环境变量文件，后缀一定要为pth
    在里面加入.py文件所在目录即可

同时，py文件中要加入

#!/usr/bin/env python
    # -*- coding: utf-8 -*-
    caffe_root = '/your/caffe/'
    import sys
    sys.path.insert(0, caffe_root + 'python')
    import caffe

不然进入`python`环境，`import caffe`成功，但执行调用该py文件的sh文件会报错说`no module named caffe`

这样就实现自己写python layer并调用成功

[py]: https://www.jianshu.com/p/e05d1b210fcb

发表评论取消回复

表情：

评论列表（有 0 条评论，197人围观）

还没有评论，来说两句吧...

相关阅读

相关 Caffe中的crop_layer层

Crop\_layer是在Fully Convolutional Networks for Semantic Segmentation这篇论文中接触到的。我发现网络配置文件中使

我会带着你远行/ 2022年12月24日 02:52/ 0 赞/ 206 阅读

相关 Caffe中的crop_layer层

Crop\_layer是在Fully Convolutional Networks for Semantic Segmentation这篇论文中接触到的。我发现网络配置文件中使

一时失言乱红尘/ 2022年11月20日 00:58/ 0 赞/ 206 阅读

相关 caffe的layer

下图是caffe layer的整体构架 ![20170104100531978][] layer整体概述 [Caffe源

Love The Way You Lie/ 2022年07月13日 09:15/ 0 赞/ 176 阅读

相关 Caffe中的特殊layer解释

slice：在某一个维度，按照给定的下标，blob拆分成几块。比如要拆分channel，总数50，下标为10,20,30,40，那就是分成5份，每份10个channel，输出5

梦里梦外;/ 2022年06月05日 11:52/ 0 赞/ 226 阅读

相关 Caffe中的crop_layer层

Crop\_layer是在Fully Convolutional Networks for Semantic Segmentation这篇论文中接触到的。我发现网络配置文件中使

Love The Way You Lie/ 2022年05月29日 07:43/ 0 赞/ 329 阅读

相关【caffe】Layer解读之：Reshape

Layer type: Reshape 头文件位置：./include/caffe/layers/reshape\_layer.hpp CPU 执行源文件位置

叁歲伎倆/ 2022年05月17日 12:39/ 0 赞/ 202 阅读

相关【caffe】Layer基类

Layer 功能：是所有的网络层的基类，其中，定义了一些通用的接口，比如前馈，反馈，reshape，setup等。 ifndef CAFFE_LAYER_H_

╰半夏微凉°/ 2022年05月16日 07:12/ 0 赞/ 203 阅读

相关【Caffe】：caffe中用的各个类型的Layer解读

构建caffe模型的时候，我们需要将模型框架写成一个protocol buffer的定义文件（prototxt）。caffe中的层以及层的参数都被定义在文件caffe.pro

Bertha 。/ 2022年05月16日 05:52/ 0 赞/ 180 阅读

相关 caffe中加入python layer的方法

小鱼趁这个机会，学学python layer 的编写及实现要想自己定义网络层的实现，如果c++代码够强大，可以在caffe底层编写cpp文件，进行caffe的编译但

谁借莪１个温暖的怀抱￠/ 2022年05月16日 02:25/ 0 赞/ 198 阅读

相关 caffe rpn layer 中的 reshape layer

Reshape层：(改变blob的形状，N,C,W,H) layer { name: "reshape" type: "Resh...

你的名字/ 2021年04月01日 19:30/ 0 赞/ 669 阅读