我已经创建了一个用于数据增强的python层,它可以很好地处理数字,但当我在ubuntu 14.04上使用终端命令训练网络时,我得到了这个错误:
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层,但我认为链接这些层有问题,我无法修复这些错误。我不能使用数字,因为我的数据是高光谱的,而数字可以接受灰度或RGB图像。
任何帮助都将不胜感激。
发布于 2016-11-30 22:58:16
根据您的prototxt文件,您应该能够运行"from digits_python_layers import AugmentationLayer“。这能(从任何目录)工作吗?
老生常谈:
您的新层应该为其层类型返回"Split“以外的值(通过其type()函数)。
https://stackoverflow.com/questions/40890538
复制相似问题