簡體   English   中英

在Caffe中創建新圖層

[英]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_CLASSREGISTER_LAYER_CLASS

關於caffe.proto :如果您希望圖層具有特定的my_layer_param消息,則只需要在那里進行更改。 如果您的圖層重復使用現有圖層的參數,則根本不需要更改caffe.proto

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM