caffe layer的type问题

各位大神好,最近在研究一份caffe的代码,我想请问一下在train.prototxt中输入层的layer中type:Imagelabelmapdata,这个type应该是自己新建的一个layer,我在src/caffe/layers中找到了image_labelmap_data.cpp文件,但是我看到网上的教程如果要增加新的layer要修改caffe.proto文件中的layertype,但是我并没有在layertype中找到这个新层,请问是哪里出了问题?还是有别的办法让这个type能够找到cpp文件。本人刚入门caffe,希望各位大神不吝赐教
已邀请:

青枫桥影

赞同来自: lawhca

如果这个layer没有参数的话,是可以不在caffe.proto文件中声明的,比如说EuclideanLoss。这时候需在cpp文件中添加两行代码
INSTANNTIATE_CLASS(EuclideanLossLayer);
REGISSTER_LAYER_CLASS(EuclideanLoss);
具体详情请参考源码,希望对你有所帮助。

要回复问题请先登录注册