caffe调试:registry失败

来源:互联网 发布:星之海洋 知乎 编辑:程序博客网 时间:2024/05/22 00:48

防止某一层未注册:

也就是下面的:

I0808 16:09:36.428869 12912 layer_factory.hpp:77] Creating layer dataF0808 16:09:36.428869 12912 layer_factory.hpp:81] Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer type: MemoryData (known types: )*** Check failure stack trace: ***
  • 1
  • 2
  • 3

只要加上以下就好,适当调整。

#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);  }
原创粉丝点击