如何找到pycaffe中使用的不同类型的图层?

时间:2016-08-22 05:50:41

标签: pycaffe

我应该在哪里查找pycaffe中可用的各种图层的文档或提示?和他们的属性?
我想定义一个使用批量标准化的体系结构,但我丢失了,在示例中只有卷积,完全连接,最大池和relu层。而已。

1 个答案:

答案 0 :(得分:1)

Pycaffe只是caffe的包装器/接口。 检查here以获取caffe中可用的图层列表。 找到您感兴趣的图层(例如batch_norm_layer.hpp)。打开头文件并检查方法type()。此方法的返回值(例如"BatchNorm")就是您要查找的内容,并在pycaffe中使用。

阅读每个头文件顶部的注释。它解释了层细节及其输入参数。 作为补充方法,请查看layer implementations。检查LayerSetUp方法并查找paramthis->layer_param_。您可以找到确切的参数及其名称。 参数的名称与网络描述(.prototxt)文件中的名称相同。找一个例子,你会找出参数。