Преобразование org.bytedeco.javacpp.Mat в массив Java int/float

Мое изображение представлено как org.bytedeco.javacpp.Mat. И я просто хочу преобразовать его в массив Java float/int. Причина преобразования заключается в том, что я хочу использовать массив Java в другой библиотеке (Nd4j) для целей перестановки изображений. Я попробовал следующие подходы, но они не работают.

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 ответ

Решение

Наиболее эффективный способ, вероятно, примерно такой:

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);
Другие вопросы по тегам