在Caffe中创建新图层

Question

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

（1）添加层以原/ caffe.proto来更新下一个可用的ID

（2）通过添加它来layer_factory.cpp使层可创建

我的查询是caffe.proto中的新ID的用途和layer_factory.cpp中的createable的功能。

在layer_factory.cpp中有一些图层可以创建。这些图层之间的区别是什么，并且不包含在layer_factory.cpp中。

Answer 1

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

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