Плохая производительность при создании INDArray с использованием DeepLearning4J

Question

Как часть большего кода, я использую это

INDArray imageArray = Nd4j.create(rgbValues);

создать мой входной массив для получения вывода из моей уже обученной нейронной сети. rgbValues ​​- это одномерный массив из примерно 10000 операций с плавающей точкой. Эта строка кода занимает около 3 секунд, и я не уверен, почему. Любая помощь с этим приветствуется.

Answer 1

Мы ответили на это в 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.