在Caffe中创建新层

Question

3

我使用 caffe 进行深度学习。要为 caffe 创建一个新层，我需要：

（1）将层添加到 proto/caffe.proto 中以更新下一个可用的ID。

（2）通过将其添加到 layer_factory.cpp 中使层可创建。

我的问题是 caffe.proto 中新 ID 的目的和 layer_factory.cpp 中 createable 的功能是什么。 layer_factory.cpp 中有些层未包含在 createable 中。那些包含和不包含在 layer_factory.cpp 中的层之间有什么区别？

- batuman

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Shai · Accepted Answer

如果您想在caffe中添加新图层，请按照wiki/Development中的说明操作。
您将看到，您不需要更改layer_factory.cpp，而是在您的cpp文件中添加两个宏INSTANTIATE_CLASS和REGISTER_LAYER_CLASS。

关于caffe.proto：仅当您希望您的图层具有特定的my_layer_param消息时，才需要在那里进行更改。如果您的图层重用现有图层的参数，则根本不需要更改caffe.proto。