本文分享了在深度学习框架Caffe中实现自定义层的详细步骤，包括编写头文件、源代码文件、修改配置文件及编译过程等。特别强调了在Python环境中正确引用新编译的pycaffe的重要性。
到头来还是改caffe的输入简单。
又调研了几天还是要写一个新层，记录一下网上查到的步骤：
1、写.hpp
2、写.cpp和.cu，注意.cpp中需要注册layer_factory
3、改.proto文件，包括层的ID，参数ID，有参数需要添加message
4、用.proto重新生成.pb.cc .pb.h .py文件
5、编译
6、如果用到python接口需要编译pycaffe，并且正确引用新的pycaffe
距离deadline还有：54天
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
