caffe添加自己编写的Python层

太过爱你忘了你带给我的痛 2022-02-23 07:58 230阅读 0赞

# 1. 前言 #

由于Python的灵活性，我们在caffe中添加自己定义的层时使用python层会更加方便，开发速速也会比C++更快，要使用python层需要编译caffe的时候打开python接口：

WITH_PYTHON_LAYER:=1

这里使用一个实现一个简单caffe python module：Data\_Process作为实验，其添加过程在后面的内容中介绍。

# 2. 实现 #

## 2.1 module Python脚本 ##

编写module的python脚本，其格式如下：

# -*- coding=utf-8 -*-
    # add caffe python layer demo
    import numpy as np
    import yaml
    import sys
    sys.path.insert(0, '../codes/caffe/python')
    import caffe
    
    
    # model class
    class Data_Process(caffe.Layer):
        def setup(self, bottom, top):
            # parse the layer parameter string, which must be valid YAML
            layer_params = yaml.load(self.param_str)
    
            # two method get params
            self.alpha = layer_params['alpha']
            self.beta = layer_params.get('beta', 0)
    
            print('Data_process Layer param:\n alpha:{}, beta:{}'.format(self.alpha, self.beta))
    
        def reshape(self, bottom, top):
            top[0].reshape(*bottom[0].shape)
    
        def forward(self, bottom, top):
            top[0].data[...] = self.alpha*bottom[0].data[...] + self.beta
    
        def backward(self, bottom, top):
            pass

## 2.2 网络定义 ##

# add python layer
    layer {
      name: "python_layer_demo"
      type: "Python"
      bottom: "Data1"
      top: "Data1_process"
      python_param {
        module: 'python_module.python_module_demo'
    	layer: 'Data_Process'
        param_str: "{'alpha': 1.01, 'beta': 0.5}"
      }
    }

# 3. 结果 #

![在这里插入图片描述][20190407141751124.png]

[20190407141751124.png]: /images/20220223/5b9b19ea82b34ea6a2efcff18a1806a8.png