使用DeepLearning4J导致INDArray创建性能不佳

时间:2017-01-03 01:40:05

标签: java arrays deeplearning4j nd4j

作为更大代码的一部分,我正在使用此

INDArray imageArray = Nd4j.create(rgbValues);

创建我的输入数组,用于从我已经训练过的神经网络中获取输出。 rgbValues是一个大约10,000个浮点数的单维数组。这一行代码大约需要3秒才能运行,我不知道为什么。对此有任何帮助表示赞赏。

1 个答案:

答案 0 :(得分:0)

我们在gitter中回答了这个问题,但我会在这里回答后人的问题。使用datavec:http://deeplearning4j.org/DataVec

我们的NativeImageLoader使用了javacv。 请参阅:https://github.com/deeplearning4j/dl4j-examples/blob/e15a0d2a87d0a203a5c202c44e6c1e9379a15e10/dl4j-examples/src/main/java/org/deeplearning4j/examples/dataExamples/MnistImagePipelineLoadChooser.java,例如用法。