原型模板
好的,现在你设计了你的图层! 这是你在 .prototxt 文件中定义它的方法 :
layer {
name: "LayerName"
type: "Python"
top: "TopBlobName"
bottom: "BottomBlobName"
python_param {
module: "My_Custom_Layer_File"
layer: "My_Custom_Layer_Class"
param_str: '{"param1": 1,"param2":True, "param3":"some string"}'
}
include{
phase: TRAIN
}
}
重要说明:
- type 必须是 Python ;
- 你必须有一个至少包含模块和图层参数的 python_param 字典 ; **** ****
- 模块是指你实现图层的文件(没有 .py );
- layer 指的是你的类的名称;
- 你可以使用 param_str 将参数传递给图层 (更多关于访问它们的信息);
- 就像任何其他图层一样,你可以定义你希望它处于活动状态的阶段(请参阅示例以了解如何检查当前阶段);