Android 人脸识别之人脸注册

mBitmap = Application.decodeImage(mFilePath);
//创建字节数组 大小由拍照传来的图片尺寸决定
byte[] data = new byte[mBitmap.getWidth() * mBitmap.getHeight() * 3 / 2];
try {
    //将bitmap转换成nv21，结果保存到data数组中
    ImageConverter convert = new ImageConverter();
    convert.initial(mBitmap.getWidth(), mBitmap.getHeight(), ImageConverter.CP_PAF_NV21);//此处高度不能为奇数
    if (convert.convert(mBitmap, data)) {
        Log.d(TAG, "convert ok!");
    }
    convert.destroy();
} catch (Exception e) {
    e.printStackTrace();
}

//创建FD人脸检测引擎
AFD_FSDKEngine engine = new AFD_FSDKEngine();
AFD_FSDKVersion version = new AFD_FSDKVersion();
List<AFD_FSDKFace> result = new ArrayList<AFD_FSDKFace>(); //人脸探测结果（探测引擎可以识别图片中的全部人联系信息，所以此处是一个List）
//初始化引擎
AFD_FSDKError err = engine.AFD_FSDK_InitialFaceEngine(
        FaceDB.appid, FaceDB.fd_key, AFD_FSDKEngine.AFD_OPF_0_HIGHER_EXT, 16, 300);
if (err.getCode() != AFD_FSDKError.MOK) {
    //引擎初始化失败
    Message reg = Message.obtain();
    reg.what = MSG_CODE;
    reg.arg1 = MSG_EVENT_FD_ERROR;
    reg.arg2 = err.getCode();
    mUIHandler.sendMessage(reg);
}
err = engine.AFD_FSDK_GetVersion(version);
//FD人脸探测，转化的nv21数据数组，传入图片的宽度、高度、NV21、探测结果
err  = engine.AFD_FSDK_StillImageFaceDetection(data, mBitmap.getWidth(), mBitmap.getHeight(), AFD_FSDKEngine.CP_PAF_NV21, result);

if (!result.isEmpty()) {
    //探测结果不为空-存在人脸，初始化FR人脸识别引擎
    AFR_FSDKVersion version1 = new AFR_FSDKVersion();
    AFR_FSDKEngine engine1 = new AFR_FSDKEngine();
    AFR_FSDKFace result1 = new AFR_FSDKFace(); //人脸特征实例
    AFR_FSDKError error1 = engine1.AFR_FSDK_InitialEngine(FaceDB.appid, FaceDB.fr_key);
    if (error1.getCode() != AFD_FSDKError.MOK) {
        //人脸识别引擎初始化失败
        Message reg = Message.obtain();
        reg.what = MSG_CODE;
        reg.arg1 = MSG_EVENT_FR_ERROR;
        reg.arg2 = error1.getCode();
        mUIHandler.sendMessage(reg);
    }
    error1 = engine1.AFR_FSDK_GetVersion(version1);
    //提取人脸识别特征 传入值为：传入的图片（NV21转换后）、图片的宽度、高度、格式、人脸检测结果列表中取出的人脸Rect、角度、提取出的人脸特征
    error1 = engine1.AFR_FSDK_ExtractFRFeature(data, mBitmap.getWidth(), mBitmap.getHeight(), AFR_FSDKEngine.CP_PAF_NV21, new Rect(result.get(0).getRect()), result.get(0).getDegree(), result1);
    if(error1.getCode() == error1.MOK) {
        //提取出了特征
        mAFR_FSDKFace = result1.clone(); //clone提取出的人脸特征
        int width = result.get(0).getRect().width();
        int height = result.get(0).getRect().height();
        Bitmap face_bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.RGB_565);//人脸位置Rect的bitmap
        Canvas face_canvas = new Canvas(face_bitmap);
        face_canvas.drawBitmap(mBitmap, result.get(0).getRect(), new Rect(0, 0, width, height), null); //将检测到的人脸位置图片提取到face_bitmap中
        Message reg = Message.obtain();
        reg.what = MSG_CODE;
        reg.arg1 = MSG_EVENT_REG;
        reg.obj = face_bitmap;
        mUIHandler.sendMessage(reg);
    } else {
        //没有提取出特征
        Message reg = Message.obtain();
        reg.what = MSG_CODE;
        reg.arg1 = MSG_EVENT_NO_FEATURE;
        mUIHandler.sendMessage(reg);
    }
    error1 = engine1.AFR_FSDK_UninitialEngine();//结束人脸识别FR引擎
} else {
    //人脸检测结果为空，图片中不存在人脸
    Message reg = Message.obtain();
    reg.what = MSG_CODE;
    reg.arg1 = MSG_EVENT_NO_FACE;
    mUIHandler.sendMessage(reg);
}
err = engine.AFD_FSDK_UninitialFaceEngine(); //结束人脸检测FD引擎

if (msg.arg1 == MSG_EVENT_REG) {
    //人脸特征信息识别成功，弹出一个对话框，输入该特征的注册名字（关联的人员信息，此处根据业务需求处理）
//**********省略*********
//****关键代码****添加人脸结果 用名字作为key
((Application)RegisterActivity.this.getApplicationContext())
    .mFaceDB.addFace(mEditText.getText().toString(), mAFR_FSDKFace);
//**********省略*********
} 

//addface
public  void addFace(String name, AFR_FSDKFace face) {
    try {
        //检查全部已注册的人脸特征信息
        boolean add = true;//默认需要注册新增人员
        for (FaceRegist frface : mRegister) {
            if (frface.mName.equals(name)) {
                //存在该人员的信息，直接向其特征list中添加新的特征
                frface.mFaceList.add(face);
                add = false;
                break;
            }
        }
        if (add) { //该人员信息没有注册过
            FaceRegist frface = new FaceRegist(name); //创建一个人员人脸特征类
            frface.mFaceList.add(face);//添加一个人脸特征
            mRegister.add(frface);
        }
        //saveInfo()方法会清空原有txt文件，重新向其中添加sdk版本信息
        if (saveInfo()) {
            //update all names
            //把当前全部数据的人员名称重新添加到txt文件
            FileOutputStream fs = new FileOutputStream(mDBPath + "/face.txt", true);
            ExtOutputStream bos = new ExtOutputStream(fs);
            for (FaceRegist frface : mRegister) {
                bos.writeString(frface.mName);
            }
            bos.close();
            fs.close();
            //将人脸特征信息写入.data文件中
            fs = new FileOutputStream(mDBPath + "/" + name + ".data", true);
            bos = new ExtOutputStream(fs);
            bos.writeBytes(face.getFeatureData());
            bos.close();
            fs.close();
        }
    } catch (FileNotFoundException e) {
        e.printStackTrace();
    } catch (IOException e) {
        e.printStackTrace();
    }
}