到头来还是改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天
定制Caffe新层实践
本文分享了在深度学习框架Caffe中实现自定义层的详细步骤,包括编写头文件、源代码文件、修改配置文件及编译过程等。特别强调了在Python环境中正确引用新编译的pycaffe的重要性。

371

被折叠的 条评论
为什么被折叠?



