原型模板

好的,现在你设计了你的图层! 这是你在 .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 将参数传递给图层 (更多关于访问它们的信息);
  • 就像任何其他图层一样,你可以定义你希望它处于活动状态的阶段(请参阅示例以了解如何检查当前阶段);