如何开始为我的图层编写代码?

时间:2017-10-11 07:04:58

标签: deep-learning caffe pycaffe matcaffe

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

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

我的问题是:

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

2)上述哪种添加新层的方法是优选的?

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

非常感谢

1 个答案:

答案 0 :(得分:2)

出于研究目的,对于“玩游戏”,编写python图层通常更方便:为您节省编译等的喧嚣。 您可以在"Python"图层here上找到简短的教程。

另一方面,如果您想获得更好的性能,您应该为您的图层编写本机c ++代码 您可以找到关于它的简短说明here