作为更大代码的一部分,我正在使用此
INDArray imageArray = Nd4j.create(rgbValues);
创建我的输入数组,用于从我已经训练过的神经网络中获取输出。 rgbValues是一个大约10,000个浮点数的单维数组。这一行代码大约需要3秒才能运行,我不知道为什么。对此有任何帮助表示赞赏。
答案 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,例如用法。