到头来还是改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天