我已经创建了一个用于数据扩充的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图像。
任何帮助都将不胜感激。
答案 0 :(得分:1)
根据您的原型文件,您应该能够“从digits_python_layers导入AugmentationLayer”运行。这是否有效(来自任何目录)?
旧答案:
您的新图层应为其图层类型(通过其type()
函数)返回“拆分”以外的内容。