原型模板

好的，現在你設計了你的圖層！ 這是你在 .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 將引數傳遞給圖層 （更多關於訪問它們的資訊）;
• 就像任何其他圖層一樣，你可以定義你希望它處於活動狀態的階段（請參閱示例以瞭解如何檢查當前階段）;