[英]Creating new layer in Caffe
我用caffe來深入學習。 要為caffe創建一個新圖層,我需要
(1)將圖層添加到proto / caffe.proto以更新下一個可用ID
(2)通過將圖層添加到layer_factory.cpp來創建圖層
我的查詢是caffe.proto中新目的的用途和layer_factory.cpp中createable的功能。
在layer_factory.cpp中有一些不包含為創建的圖層。 包含和不包含在layer_factory.cpp中的這些層之間有什么區別。
如果要為caffe添加新圖層,請按照wiki / Development中的說明進行操作。
您將看到您不需要更改layer_factory.cpp
,而是在cpp文件中添加兩個宏INSTANTIATE_CLASS
和REGISTER_LAYER_CLASS
。
關於caffe.proto
:如果您希望圖層具有特定的my_layer_param
消息,則只需要在那里進行更改。 如果您的圖層重復使用現有圖層的參數,則根本不需要更改caffe.proto
。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.