caffe-windows添加层

最近接触到Fine-Grained Classification, 其中的经典方法Bilinear CNN用到了Bilinear Layer, L2-normlize Layer, signed-sqrt Layer等

这些层在caffe-windows已经BVLC版caffe中是不存在的,若想应用的话需要将上述层添加到caffe工程

https://github.com/gy20073/compact_bilinear_pooling/tree/master/caffe-20160312 好在已经有大神们做了这些层的相关implementation.. 感谢

 

对于Linux环境下caffe的话之前做过相关层的添加,过程比较简单安装上述链接中的步骤走完即可,大体步骤:

1、src 中的cpp及cu文件复制到自己caffe工程相对的src中

2、include中的hpp文件复制到自己caffe相应的include中

3、src中 caffe.proto 文件中添加新层的参数,注意序号不能与已存在层重复

因为BCNN还涉及到了cuda中的相关计算,所以还做了cuda的相关修改,见链接

 

对于caffe-windows则需要将步骤1,2中的文件在vs的sln文件中添加到相应的位置..否则生成的工程中还是没有这些层的应用

posted on 2018-08-07 22:32  klitech  阅读(336)  评论(0编辑  收藏  举报

导航