Android JNI-加载dlib形状预测器文件

时间:2018-10-02 14:26:17

标签: java android android-ndk java-native-interface dlib

在正常的Visual Studio C ++项目中,我使用以下代码将形状预测器文件加载到dlib shape_predictor变量中:

    dlib::shape_predictor spredictor;
    //calling the file with a relative path
    dlib::deserialize("sp68fl.dat") >> spredictor;

但是,在Android中使用JNI,我从我的活动中加载了此文件,并将其作为字节数组传递。因此,我有一个jbyteArray,它是用于使用Dlib进行人脸检测的shape_predictor_68_face_landmarks.dat文件。

我发现一个重载的方法deserialize(serializable_type&item,std :: istream&in),但不知道如何使用它(也许我必须将jbyteArray转换为该istream)。

如何从jbyteArray将此文件加载到dlib :: shape_predictor?

编辑:

这是我到目前为止尝试过的:

#include <jni.h>
#include <dlib\image_processing.h>
#include <dlib\image_processing\frontal_face_detector.h>
#include <dlib\image_processing\render_face_detections.h>
#include <dlib\image_processing\generic_image.h>
#include <dlib\image_processing\full_object_detection.h>
#include <dlib\opencv.h>
#include <dlib\geometry.h>
#include <dlib\image_transforms.h>
#include <dlib\serialize.h>
#include <dlib\gui_widgets.h>
#include <dlib\image_io.h>

extern "C" {
struct membuf : std::streambuf
{
    membuf(char* begin, char* end) {
        this->setg(begin, begin, end);
    }
};


JNIEXPORT jstring JNICALL
Java_br_edu_infnet_test16_MainActivity_stringFromJNI(JNIEnv *env, jobject daobj,
                                                     jbyteArray fileBytes,
                                                     jint fileSize) {
    dlib::shape_predictor spredictor;

    jboolean isCopy = true;
    jbyte * a = env->GetByteArrayElements(fileBytes, &isCopy);
    char* teste = (char *) a;

    std::ifstream in;
    //stream is not loading? stream size is zero after read method
    in.read(teste, fileSize);

    try {
        dlib::deserialize(spredictor, in);
    } catch (dlib::serialization_error &serializationError) {
        //deserialize method fails. Dlib error: deserializing object of type int
        std::cout << "ERROR: " << serializationError.what();
    }

    in.close();
    env->ReleaseByteArrayElements(fileBytes, a, JNI_ABORT);

    const char *cppString = "Diego";
    jstring javaString = env->NewStringUTF(cppString);
    return javaString;
}
}

发送字节数组的活动:

package br.edu.infnet.test16;

import android.os.Bundle;
import android.support.v7.app.AppCompatActivity;
import android.util.Log;
import android.widget.TextView;

import java.io.InputStream;

public class MainActivity extends AppCompatActivity {

    // Used to load the 'native-lib' library on application startup.
    static {
        System.loadLibrary("native-lib");
    }

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        // Example of a call to a native method
        TextView tv = findViewById(R.id.sample_text);
//        tv.setText(stringFromJNI());
        byte[] buffer = null;
        try {
            InputStream inputStream = getAssets().open("sp68fl.dat");
            int size = inputStream.available();
            buffer = new byte[size];
            inputStream.read(buffer);
            inputStream.close();
            stringFromJNI(buffer, size);
        } catch (Exception e) {
            e.printStackTrace();
            Log.e("ERRO", "DEU EEEEEEERRO: " + e.getMessage());
        }
    }

    public native String stringFromJNI(byte[] fileBytes, int fileSize);
}

看起来像问题在于ifstream没有使用变量teste(一个char指针)加载。该文件的大小约为100MB。

2 个答案:

答案 0 :(得分:3)

您可以用C ++打开文件。问题是,在Android中,您需要文件名的完整路径,没有可依赖的当前目录。使用Java API准备完整路径并将字符串传递给C ++而不是发送缓冲区可能会更容易。

如果不可能,则必须将 jbyteArray 转换为 char [] (请参见 How to convert jbyteArray to native char* in jni? ),然后创建一个 std :: istream ,如 Get an istream from a char*

答案 1 :(得分:0)

我遇到了同样的问题,我只获得了在存储上读写的权限,并且可以正常工作。 尝试添加清单:

并要求获得许可

requestPermissions(新字符串[] {Manifest.permission.WRITE_EXTERNAL_STORAGE,Manifest.permission.READ_EXTERNAL_STORAGE},REQUEST_PERMISSION);

我将文件保存在root上,并使用了绝对路径来定位。