将org.bytedeco.javacpp.Mat转换为Java int / float数组

时间:2016-03-03 03:52:11

标签: java javacpp deeplearning4j nd4j dl4j

我的图片表示为org.bytedeco.javacpp.Mat。我只想将它转换为float / int的Java数组。转换的原因是我想在其他库(Nd4j)中使用Java数组来进行图像置换。我尝试了以下方法,但它们不起作用。

private static int[] MatToFloatArray1(Mat mat) {
        org.bytedeco.javacpp.BytePointer matData = mat.data();
        byte[] d = new byte[matData.capacity()];
        return toIntArray(d);
    }


private static int[] MatToFloatArray2(Mat mat) {
    org.bytedeco.javacpp.BytePointer matData = mat.data();
    IntBuffer intBuffer = matData.asBuffer().asIntBuffer();
    return intBuffer.array();

}
    private static int[] toIntArray(byte[] d) {
        IntBuffer intBuf =
                ByteBuffer.wrap(d)
                        .order(ByteOrder.BIG_ENDIAN)
                        .asIntBuffer();
        int[] array = new int[intBuf.remaining()];
        return array;

}

1 个答案:

答案 0 :(得分:1)

最有效的方法可能如下:

Mat intMat = new Mat();
mat.convertTo(intMat, CV_32S);
IntBuffer intBuffer = intMat.createBuffer();
int[] intArray = new int[intBuffer.capacity()];
intBuffer.get(intArray);

这是int的情况,但我们也可以为float做同样的事情:

Mat floatMat = new Mat();
mat.convertTo(floatMat, CV_32F);
FloatBuffer floatBuffer = floatMat.createBuffer();
float[] floatArray = new float[floatBuffer.capacity()];
floatBuffer.get(floatArray);