您的位置:首页 > 理论基础 > 计算机网络

【神经网络与深度学习】Caffe训练执行时爆出的Check failed: registry.count(t ype) == 1 (0 vs. 1) Unknown layer type

2016-09-14 23:20 991 查看
 自己建立一个工程,希望调用libcaffe.lib ,各种配置好,也能成功编译,但是运行就会遇到报错

F0519 14:54:12.494139 14504 layer_factory.hpp:77] Check failed: registry.count(t ype) == 1 (0 vs. 1) Unknown layer type: Input (known types: Input )

   各种跟踪调试都找不到原因,相同的代码在microsoft版本的caffe里使用就没有问题,自己新建一个项目就出错。

   网上找了一下原因,大概是libcaffe.lib是静态库的方式编译的,有一些东西没法使用,暂时只能肤浅的这么理解。我试过改成动态库,编译都通不过了。

一些参考方案:
http://stackoverflow.com/questions/30325108/caffe-layer-creation-failure http://stackoverflow.com/questions/32416226/create-layer-error-at-layer-factory-hpp https://groups.google.com/forum/#!topic/caffe-users/Py6IwMQvtqo/discussion
尝试了不少,都不怎么给力。

我甚至在github项目下提问



官方答复是只能在解决方案里面新建项目。

当然,最后我找到了一个办法,



一开始没有太看明白这是什么意思。

  后来仔细看了一下libcaffe工程的目录结构,发现有个layer目录,里面放的都是各种层,什么ArgMaxLayer,ScaleLayer,当然也有InputLayer,感觉很有可能问题就出在这些代码里面,仔细阅读input_layer.cpp,果然发现有一句

INSTANTIATE_CLASS(InputLayer);  

REGISTER_LAYER_CLASS(Input); 

所以解决方案就是,新建一个头文件,例如head.h,将他提示你没有的层的头文件包含进去,在添加类似于

extern INSTANTIATE_CLASS(InputLayer);的这句话

后来发现有的层添加了也会报错,那是因为他们没有类似于REGISTER_LAYER_CLASS(Input);这句话。可以想象,REGISTER_LAYER_CLASS这句话作用就是注册layer。

可以添加进去。

我贴上我的head.h,添加了我使用到的layer

#include "caffe/common.hpp"  

#include "caffe/layers/input_layer.hpp"  

#include "caffe/layers/inner_product_layer.hpp"  

#include "caffe/layers/dropout_layer.hpp"  

#include "caffe/layers/conv_layer.hpp"  

#include "caffe/layers/relu_layer.hpp"  

  

#include "caffe/layers/pooling_layer.hpp"  

#include "caffe/layers/lrn_layer.hpp"  

#include "caffe/layers/softmax_layer.hpp"  

  

  

namespace caffe  

{  

  

    extern INSTANTIATE_CLASS(InputLayer);  

    extern INSTANTIATE_CLASS(InnerProductLayer);  

    extern INSTANTIATE_CLASS(DropoutLayer);  

    extern INSTANTIATE_CLASS(ConvolutionLayer);  

    REGISTER_LAYER_CLASS(Convolution);  

    extern INSTANTIATE_CLASS(ReLULayer);  

    REGISTER_LAYER_CLASS(ReLU);  

    extern INSTANTIATE_CLASS(PoolingLayer);  

    REGISTER_LAYER_CLASS(Pooling);  

    extern INSTANTIATE_CLASS(LRNLayer);  

    REGISTER_LAYER_CLASS(LRN);  

    extern INSTANTIATE_CLASS(SoftmaxLayer);  

    REGISTER_LAYER_CLASS(Softmax);  

  

}  

---

added by ZhangPY 2016-9-14 23:21

注意,这里我的调试过程中还出现了注册过再次注册的错误,但是去掉自己添加的注册后,又爆出没有注册,所以我索性直接将caffelib源码中的layer.hpp中的注册部分给注释掉了,然后用我自己的地方去注册它,这样就不再出现问题了。

你所使用的layer可能跟我不一样,但是解决方法是一样的,提示哪个layer找不到,就把那个layer添加进来就是了。

彻底一点的方案是把所有的layer全写进来。

   我在添加过程中,之前在每个extern INSTANTIATE_CLASS 后面都添加 REGISTER_LAYER_CLASS,结果报错,说重复注册。于是删掉了。

所以,如果你以后遇到没有注册的layer,就照上面做就好了,万一遇到重复注册,也应该会举一反三,知道如何解决,例如把判断重复注册的语句修改一下等等。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: