在 Caffe 中创建新层

Question

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

(1)向proto/caffe.proto 添加层以更新下一个可用ID

(2)通过将图层添加到 layer_factory.cpp

使其可创建

我的问题是 caffe.proto 中新 ID 的用途是什么以及 layer_factory.cpp 中 createable 的功能。

layer_factory.cpp 中有一些图层未作为可创建图层包含在内。 layer_factory.cpp.

中包括和不包括的那些层有什么区别

Answer 1

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

关于caffe.proto：如果您希望您的层具有特定的my_layer_param消息，您只需要更改那里。如果您的层重复使用现有层的参数，则根本不需要更改 caffe.proto。