首页 > 代码库 > 如何向caffe中添加层
如何向caffe中添加层
修改caffe\src\caffe\proto\caffe.proto
在最后添加centerloss层的参数信息
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]; }
添加层类型信息,注意不能与其他层的数字相同。
optional CenterLossParameter center_loss_param = 148;
将cpu和gpu两个模式下运行的文件添加到src\caffe\layers\
如何编写层,请查看官方文档或者谷歌
3.添加层的头文件
向\include\caffe\layers\中添加头文件
如何向caffe中添加层
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。