首页 > Windows开发 > 详细

caffe-windows添加层

时间:2018-08-07 22:41:34      阅读:261      评论:0      收藏:0      [点我收藏+]

最近接触到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文件中添加到相应的位置..否则生成的工程中还是没有这些层的应用

caffe-windows添加层

原文:https://www.cnblogs.com/klitech/p/9439949.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!