2016-01-27 41 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

相关问题