Caffe运行自带Demo出现Unknown layer type错误解决

最新推荐文章于 2022-01-04 21:11:08 发布

shenck1992

最新推荐文章于 2022-01-04 21:11:08 发布

阅读量8.4k

点赞数

分类专栏：使用记录

本文链接：https://blog.csdn.net/shenck1992/article/details/49452245

版权

当尝试运行Caffe的mnist训练脚本时，出现Unknown layer type: Data错误。该问题源于layer_factory中未正确注册Data层。解决方案是修改Makefile，将-Wl,--whole-archive选项添加到链接命令，确保所有层的代码被编译进caffe.bin，避免链接时丢失某些层。参考GitHub和StackOverflow上的相关讨论可找到详细解决步骤。" 131895813,14732956,中科亿海微在2023世界半导体大会展示FPGA创新成果,"['fpga开发', '业界资讯', '半导体技术', '集成电路设计', '通信技术']

摘要由CSDN通过智能技术生成

错误描述：

参见：https://github.com/BVLC/caffe/issues/1986

运行sh ./example/mnist/train_lenet.sh时出现错误：layer_factory.hpp:77] Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer type: Data (known types: )

原因分析（大概）：

Caffe中每种层都需要在layer_factory中“注册”，然后才能根据prototxt中创建出来，然后build caffe的tools时，编译命令链接到caffe的静态库.a文件，由于没有直接引用各个层，所以这些层的代码并没有从.a文件中编译到./build/tools/caffe.bin中，导致无法找到层的类型。

成因推测：

我编译次数不多，推测是使用cmake生成Makefile时的参数不对，导致编译的工具集与caffe的静态库链接。

解决办法：

修改Makefile中关于tools相关部分(注意根据情况不同修改，注意备份)：

原版：