Caffe层创建失败

克勒勒姆

我正在尝试在测试阶段加载网络配置,该网络配置首先有一个内存数据层,然后是一个卷积层。MemoryData层的创建成功,但是卷积层的创建在以下位置失败:

LOG(INFO) << "Creating layer " << param.name();
const string& type = param.type();
CreatorRegistry& registry = Registry();
CHECK_EQ(registry.count(type), 1) << "Unknown layer type: " << type
<< " (known types: " << LayerTypeList() << ")";

打印错误为:

F0519 14:54:12.494139 14504 layer_factory.hpp:77]检查失败:registry.count(t ype)== 1(0对1)未知图层类型:卷积(已知类型:MemoryData)

注册表只有一个条目,确实带有MemoryData。进入注册表创建功能时,它看起来像是第一个(最后一个,因为这是一个单调),从

REGISTER_LAYER_CLASS(MemoryData);

在memory_data_later.cpp中。

我看到REGISTER_LAYER_CLASS了对其他受支持层的类似调用,但似乎从未调用过它们。我该如何解决?

谢谢!

忽视

尝试将caffe静态链接到可执行文件时,会发生此错误。您需要传递额外的链接器标志,以确保包含层注册代码。

如果使用cmake,请查看Targets.cmake:

###########################################################################################
# Defines global Caffe_LINK flag, This flag is required to prevent linker from excluding
# some objects which are not addressed directly but are registered via static constructors
if(BUILD_SHARED_LIBS)
  set(Caffe_LINK caffe)
else()
  if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang")
    set(Caffe_LINK -Wl,-force_load caffe)
  elseif("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
    set(Caffe_LINK -Wl,--whole-archive caffe -Wl,--no-whole-archive)
  endif()
endif()

然后在哪里创建目标:

# target
add_executable(${name} ${source})
target_link_libraries(${name} ${Caffe_LINK})

一种快速的解决方案是将caffe构建和链接为共享库而不是静态库。

另请参阅这篇文章

只需在Windows上完成MSVC编译即可:在目标可执行文件或dll上使用/ OPT:NOREF/ INCLUDE链接器选项。

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章