zoukankan      html  css  js  c++  java
  • caffe win添加新层

    1、编写.h和.cpp .cu文件

    将.hpp文件放到路径caffe-windowscaffe-masterincludecaffelayers下 
    将.cpp文件和.cu放到路径caffe-windowscaffe-mastersrccaffelayers下

    修改caffe-windowscaffe-mastersrccaffeprotocaffe.proto,需要修改两处
    1.找到 LayerParameter 描述,增加一项:

    message LayerParameter {
    ...
    optional ShuffleChannelParameter shuffle_channel_param = 164;
    }

    注意,这里的参数设置不能与LayerParameter中其他参数相同


    2.仍然在 caffe.proto 中,增加 

    ShuffleChannelParameter 

    声明,位置任意。其功能是可以用于从 prototxt 中读取预设值。

    message ShuffleChannelParameter {
      optional uint32 group = 1[default = 1]; // The number of group
    }

    将我们定义的层加载到Caffe中去

    在libcaffe项目中include文件夹 layers——添加.h文件

    在libcaffe项目中src文件夹layers——添加.cpp文件

    重新编译caffe即可

  • 相关阅读:
    bzoj1009
    bzoj1576 3694
    bzoj3143
    bzoj1391
    bzoj2729
    bzoj2653
    bzoj3261
    bzoj2326
    人件
    优秀的产品
  • 原文地址:https://www.cnblogs.com/crazybird123/p/10396001.html
Copyright © 2011-2022 走看看