windows caffe添加新层center loss layer

最新推荐文章于 2019-07-26 14:28:29 发布

qq_34662616

最新推荐文章于 2019-07-26 14:28:29 发布

阅读量360

点赞数

分类专栏： caffe 文章标签： caffe

本文链接：https://blog.csdn.net/qq_34662616/article/details/81235304

版权

caffe 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

相关文件可见：https://github.com/ydwen/caffe-face

主要包括4个步骤：

（1）src\caffe\caffe.proto里添加消息机制：proto格式的ID号（要求ID号唯一，不与其他ID重合即可）及相应定义

在message LayerParameter{}中添加如下ID号：
optional CenterLossParameter center_loss_param = 155;

         再在文档最后加入：
         message CenterLossParameter {
   optional uint32 num_output = 1; // The number of outputs for the layer
         optional FillerParameter center_filler = 2; // The filler for the centers
   // The first axis to be lumped into a single inner product computation;
// all preceding axes are retained in the output.
// May be negative to index from the end (e.g., -1 for the last axis).
       optional int32 axis = 3 [default = 1];
      }

（2）include/caffe/layers/添加头文件center_loss_layers.hpp

（3）src/caffe/layers/添加CPU、GPU实现的相关文件center_loss_layers.cpp\center_loss_layers.cu

（4）重新编译Caffe。首先在windows/libcaffe/libcaffe.vcxproj和libcaffe.vcxproj.filters里

libcaffe.vcxproj相应处增加：

libcaffe.vcxproj相应处增加：

       <ClCompile Include="..\..\src\caffe\layers\center_loss_layer.cpp">
        <Filter>src\layers</Filter>
   </ClCompile>

       <ClInclude Include="..\..\include\caffe\layers\center_loss_layer.hpp">
        <Filter>include\layers</Filter>
   </ClInclude>

       <CudaCompile Include="..\..\src\caffe\layers\center_loss_layer.cu">
        <Filter>cu\layers</Filter>
   </CudaCompile>

再重新编译。成功即可使用。