在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。