1

我有 Android 本机代码（C++共享对象），希望在创建 APK 时read对保存在文件夹中的文件进行建模。assets

Tensorflow Lite 有两个 API：FromFile 和 FromBuffer

static std::unique_ptr<FlatBufferModel> BuildFromFile(const char* filename, ErrorReporter* error_reporter);

static std::unique_ptr<FlatBufferModel> BuildFromBuffer(const char* buffer, size_t buffer_size, ErrorReporter* error_reporter);

使用以下代码，我可以访问 FromBuffer：

Java代码：

private AssetManager mgr;

// Get mgr 
mgr = getResources().getAssets();

C++代码：

AAssetDir* assetDir = AAssetManager_openDir(mgr, "");
const char* filename = (const char*)NULL;
while ((filename = AAssetDir_getNextFileName(assetDir)) != NULL) {
    AAsset* asset = AAssetManager_open(mgr, filename, AASSET_MODE_STREAMING);
    char buf[BUFSIZ];
    int nb_read = 0;
    FILE* out = fopen(filename, "w");
    while ((nb_read = AAsset_read(asset, buf, BUFSIZ)) > 0)
        fwrite(buf, nb_read, 1, out);
    fclose(out);
    AAsset_close(asset);
}
AAssetDir_close(assetDir);

任何想法，如何访问assets文件夹以使用BuildFromFile？

4

1 回答 1

2

我不认为BuildFromFile不将资产文件复制到本地文件就可以使用（这相当于您在上面所做的，然后BuildFromBuffer在这种情况下更方便）。

如果您想避免内存复制，您可以执行以下操作：

  1. (Java) 通过内存映射加载模型。
    AssetFileDescriptor fileDescriptor = getResources().getAssets().openFd(filePath);\
    FileInputStream inputStream = new FileInputStream(fileDescriptor.getFileDescriptor());\
    FileChannel fileChannel = inputStream.getChannel();\
    long startOffset = fileDescriptor.getStartOffset();\
    long declaredLength = fileDescriptor.getDeclaredLength();\
    MappedByteBuffer modelBuffer = fileChannel.map(FileChannel.MapMode.READ_ONLY, startOffset, declaredLength);
  1. (Java)通过jni将映射文件传递给C++，比如定义如下方法：
    private static native void initJniWithByteBuffer(ByteBuffer modelBuffer)
  1. (C++) 使用缓冲文件来初始化你的模型：
    Java_xxxxxx_initJniWithByteBuffer(JNIEnv* env, jclass thiz, jobject model_buffer) {\
      char* buffer = static_cast<char*>(env->GetDirectBufferAddress(model_buffer));\
      size_t buffer_size = static_cast<size_t>(env->GetDirectBufferCapacity(model_buffer));\
    }
于 2020-12-16T01:10:18.080 回答