2

我看到研究人员正在向原始版本的 Caffe 添加一些功能,并根据他们的需要使用这些层和功能,然后通过Github. 如果我没记错的话,有两种方法:1)在添加 c++ 和 Cuda 版本的层之后重新编译 Caffe。2)为该功能编写 python 代码并将其称为 Caffe 中的 python 层。

我想根据我的研究问题向 Caffe 添加一个新层。我真的不知道我应该从哪一点开始编写新层以及我应该考虑哪些步骤。

我的问题是:

1) 是否有任何文档或任何学习资料可用于编写图层?

2)上述添加新层的方法中首选哪种方式?

我非常感谢任何帮助和指导

非常感谢

4

1 回答 1

2

出于研究目的,为了“玩耍”,编写一个 python 层通常更方便:省去编译等的麻烦。
你可以在这里"Python"找到关于层的简短教程。

另一方面,如果您想要更好的性能,您应该为您的层编写本机 C++ 代码。您可以在此处
找到有关它的简短说明。

于 2017-10-11T10:29:59.283 回答