如何将jpeg图像转换为矩阵秩2数组，以便模型进行预测(使用deeplearning4j库)

import org.datavec.image.loader.NativeImageLoader;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;
import org.deeplearning4j.nn.modelimport.keras.KerasModelImport;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;

public class ImageToMatrixExample {
    public static void main(String[] args) throws Exception {
        // 加载JPEG图像
        NativeImageLoader loader = new NativeImageLoader();
        INDArray image = loader.asMatrix("path/to/jpeg/image.jpg");

        // 图像预处理
        // 可以使用loader对象的resizeImage()、cropImage()等方法进行预处理操作

        // 转换为矩阵秩2数组
        INDArray matrix = image.reshape(1, -1);

        // 模型预测
        MultiLayerNetwork model = KerasModelImport.importKerasSequentialModelAndWeights("path/to/model.h5");
        INDArray output = model.output(matrix);

        System.out.println(output);
    }
}