码迷,mamicode.com
首页 > Windows程序 > 详细

caffe-windows添加层

时间:2018-08-07 22:55:59      阅读:540      评论:0      收藏:0      [点我收藏+]

标签:文件中   linux环境   自己   pool   ssi   https   .com   inux   复制   

最近接触到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添加层

标签:文件中   linux环境   自己   pool   ssi   https   .com   inux   复制   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!