Преобразование 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);