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

最新推荐文章于 2023-03-09 15:46:32 发布

weixin_30919571

最新推荐文章于 2023-03-09 15:46:32 发布

阅读量229

点赞数

文章标签：人工智能

原文链接：http://www.cnblogs.com/huty/p/8518102.html

版权

自己建立一个工程，希望调用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);

}