检查失败:registry.count(type)== 0(1 vs. 0)图层类型拆分已注册

时间:2016-11-30 14:11:34

标签: caffe pycaffe

我已经创建了一个用于数据扩充的python层,它可以很好地处理数字,但是当我在ubuntu 14.04上使用terminal命令训练网络时,我收到了这个错误:

I1130 16:29:56.155732 18230 layer_factory.hpp:77] Creating layer aug_data
F1130 16:29:56.220578 18230 layer_factory.hpp:69] Check failed: registry.count(type) == 0 (1 vs. 0) Layer type Split already registered.

其中aug_data是自定义python层。我已经在配置文件中进行了更改以接受python层,但我认为链接层无法解决问题。我不能使用DIGITS,因为我的数据是高光谱的,而DIGITS接受灰度或RGB图像。

任何帮助都将不胜感激。

1 个答案:

答案 0 :(得分:1)

根据您的原型文件,您应该能够“从digits_python_layers导入AugmentationLayer”运行。这是否有效(来自任何目录)?

旧答案:

您的新图层应为其图层类型(通过其type()函数)返回“拆分”以外的内容。

相关问题