Caffe python layer方法执行时机

最新推荐文章于 2022-11-09 20:48:37 发布

Daniel2333

最新推荐文章于 2022-11-09 20:48:37 发布

阅读量2.1k

点赞数 2

分类专栏： caffe 文章标签： caffe pycaffe

本文链接：https://blog.csdn.net/weixin_35653315/article/details/54632277

版权

caffe 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

用python定义Layer的时候, 需要实现以下几个方法:

class MyPythonLayer(caffe.Layer):
    def setup(self, bottom, top):
        pass

    def reshape(self, bottom, top):
        pass

    def forward(self, bottom, top):
        pass

    def backword(self, top, propagate_down, bottom):
        pass

各个方法都是干什么用的, 什么时候调用呢?

setup: 初始化layer, 只调用一次.
reshape : 每次forward之前都会调用一次, 用于将blobs reshape到需要的形状. (应该)是为forward数据分配内存空间. 注意, 是先调用完所有layer的reshape方法才开始调用它们的forward方法. 不要在reshape方法里处理数据的数值, 因为reshape方法运行时, bottom传入的数据并不是forward来的数据, 确切的说, 都只是分配了空间并用0填充. 在forward之前一定要有top[i].reshape调用, 假如shape不变的话, 可以在setup里面只执行一次, 否则就把top[i].reshape操作写在MyPythonLayer.reshape()方法里
forward: forward时调用
backward: backward时调用.