JNI获取assets文件夹内的文件地址
0、前言
在深度学习的JNI时,需要把模型文件放到assets里,加载时,却不知道模型的目录地址,报错:该文件未找到.
因为assert文件夹 只是Android系统管理,加载模型需要从系统文件中读取.所以需要把assert的文件写到系统文件中.
1、NCNN
ncnn已经集成了AAssetManager,在模型加载时,传入AAssetManager 参数,他就会实现自动写入系统,并读取相应目录地址,JNI例子如下:
#include <jni.h>
#include <string>
#include <iostream>
#include <net.h>
#include <android/asset_manager_jni.h> //AAssetManager需要
//模型文件mnet.25-opt.param、mnet.25-opt.bin放在asserts中
static ncnn::Net retinaface;
extern "C" JNIEXPORT jboolean JNICALL
Java_com_example_retainfacencnn_RetinaFace_Init(JNIEnv *env, jobject thiz, jobject assetManager) {
AAssetManager *mgr = AAssetManager_fromJava(env, assetManager);
int ret = retinaface.load_param(mgr, "mnet.25-opt.param");
ret = retinaface.load_model(mgr, "mnet.25-opt.bin");
AAssetManager其他功能:
AAssetManager *mgr = AAssetManager_fromJava(env, assetManager)
//打开assert文件夹
AAssetDir *dir = AAssetManager_openDir(mgr,"");
//打开assert下的子文件夹part1
AAssetDir *dir = AAssetManager_openDir(mgr,"part1");
//获取文件名
const char *file = nullptr;
file =AAssetDir_getNextFileName(dir));
//打开文件
AAsset* asset = AAssetManager_open(AAssetManager* mgr, const char* filename, int mode);
//mode 可以为AASSET_MODE_STREAMING
//获取文件长度 获取文件长度，如果文件不是很大可以直接用malloc分配空间使用AAsset_read进度文件内容读取。
int len = AAsset_getLength(asset);
//读取文件
unsigned char *buf = (unsigned char *) malloc(sizeof(unsigned char) * len);
AAsset_read(asset, buf, static_cast<size_t>(len));
//关闭文件
AAsset_close(asset);
//关闭文件夹
AAssetDir_close(dir);
2、手动获取地址
CPP无法直接获取文件目录地址,需要在JAVA中对文件进行拷贝到系统中,获取它的绝对地址,传给JNI
2.1 JAVA类方法
2.1.1 方法1
package com.ylz.seetaface;
import java.io.FileNotFoundException; //异常
import android.content.res.AssetManager;
import java.io.BufferedInputStream;
import java.io.FileOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
public class MainActivity extends AppCompatActivity {
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
String cstapath = getPath("face_detector.csta", this); //调用获取文件地址方法
boolean ret_init = seetaFace.loadModel(cstapath) //加载模型
...
(其他代码略)
...
//获取文件地址方法
private static String getPath(String file, Context context) {
AssetManager assetManager = context.getAssets();
BufferedInputStream inputStream = null;
try {
// Read data from assets.
inputStream = new BufferedInputStream(assetManager.open(file)); //打开文件放入输入流中
byte[] data = new byte[inputStream.available()]; //输入流信息放入data中
inputStream.read(data); //输入流读取data
inputStream.close(); //输入流关闭
// Create copy file in storage.
File outFile = new File(context.getFilesDir(), file); //新文件(夹)创建,应用文件目录
FileOutputStream os = new FileOutputStream(outFile); //输出流
os.write(data); //写入数据.即完成了拷贝
os.close(); //关闭
// Return a path to file which may be read in common way.
Log.d("filePath:",outFile.getAbsolutePath());
return outFile.getAbsolutePath(); //得到新文件的绝对地址
} catch (IOException ex) {
Log.i("打开失败", "无法打开此文件");
}
return "";
}
}
}
2.1.2 方法2
package com.ylz.seetaface;
import java.io.FileNotFoundException; //异常
import android.content.res.AssetManager; //获取getAssets()
import java.io.BufferedInputStream;
import java.io.FileOutputStream;
import java.io.File; //获取getCacheDir()
import java.io.IOException;
import java.io.InputStream;
public class MainActivity extends AppCompatActivity {
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
boolean mkdir_model = copyAssetAndWrite("face_detector.csta");
if(mkdir_model==true){
File dataFile=new File(getCacheDir(),"face_detector.csta");
String cstapath=dataFile.getAbsolutePath();
Log.e("文件地址为:", cstapath);
boolean ret_init = seetaFace.loadModel(cstapath);
}
...
(其他代码略)
...
//获取文件地址方法
private boolean copyAssetAndWrite(String fileName){
try {
File cacheDir=getCacheDir(); //获取文件夹 缓冲目录
if (!cacheDir.exists()){
cacheDir.mkdirs();
}
File outFile =new File(cacheDir,fileName); //新建文件
if (!outFile.exists()){
boolean res=outFile.createNewFile(); //如果不存在,则新建
if (!res){
return false; //新建失败
}
}else {
if (outFile.length()>10){//表示已经写入一次
return true;
}
}
InputStream is=getAssets().open(fileName); //输入流打开文件
FileOutputStream fos = new FileOutputStream(outFile); //新文件(这时候还是空)放入输出流
byte[] buffer = new byte[1024];
int byteCount;
while ((byteCount = is.read(buffer)) != -1) {
fos.write(buffer, 0, byteCount); //输出流写入数据
}
fos.flush();
is.close(); //输入流关闭
fos.close(); //输出流关闭
return true;
} catch (IOException e) {
e.printStackTrace();
}
return false;
}
2.2 JNI中方法
#include <jni.h>
#include <string>
#include <seeta/Common/Struct.h>
static Seetaface seetaNet;
extern "C" JNIEXPORT jboolean
Java_com_ylz_seetaface_SeetaFace_loadModel(JNIEnv *env,jobject thiz,jstring cstapath) {
//模型初始化
const char *modelpath = env->GetStringUTFChars(cstapath, 0); //jstring转char*
LOGI("model_path:%s",modelpath);
seetaNet.Init(model_path);
env->ReleaseStringUTFChars(cstapath, modelpath);
2.3 获取的地址
两种方法获取的地址区别:
方法一获取的文件地址为:/data/user/0/com.ylz.seetaface/files/face_detector.csta
com.ylz.seetaface : 应用创建的JAVA类名
files: 利用getFilesDir(),得到的是应用文件目录
方法二获取的文件地址为:/data/user/0/com.ylz.seetaface/cache/face_detector.csta
- cache: 利用getCacheDir() 得到的是应用缓存目录
区别在于子文件夹名,方法1为files, 方法2为cache
只需修改代码中的getFilesDir()或者getCacheDir() 即可保持一致
各自获取子文件夹目录地址转string:
getCacheDir().getPath();
getFilesDir().getPath();
参考1(https://www.jianshu.com/p/0727c8a5f8e4) Android JNI实现图片处理和coco-mobilenet模型加载（OpenCV）
参考2(https://blog.csdn.net/dreamsever/article/details/80468464) Android获取assets文件路径
参考3(https://www.jianshu.com/p/127490a3bdd2) [Android]JNI进阶1-读取本地文件夹
