2016-01-27 43 views
1

我想學習如何編寫caffe python圖層。
但我只找到像pyloss這樣非常簡單的圖層的例子。如何用可訓練參數編寫caffe python圖層?

如何用可訓練參數編寫python caffe?

例如,如何編寫完全連接的python圖層?

+0

[Caffe:如何獲得Python層的階段?]可能的副本(http://stackoverflow.com/questions/34549743/caffe-how-to-get-the-phase-of-a-python層) – Shai

+0

@Shai,我不認爲這兩個是同一個問題。我的問題是如何創建一個帶有參數的python圖層,例如,如何實現一個python完全連接的圖層? –

+0

另一個問題是關於用參數編寫一個python圖層 - 這正是您的問題的標題。 – Shai

回答

1

Caffe將圖層的可訓練參數存儲爲blobs的矢量。默認情況下,該矢量爲空,您可以在圖層的setup中爲其添加參數blob。在test_python_layer.py中有一個參數層的簡單示例。

有關"Python" caffe圖層的更多信息,請參閱this post

相關問題