java - 使用 DeepLearning4J 的 INDArray 创建性能不佳

Question

作为一段更大的代码的一部分，我正在使用这个

INDArray imageArray = Nd4j.create(rgbValues);

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

Answer 1

我们在 gitter 中回答了这个问题，但我会在这里为后代回答。使用数据向量： 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。