Android虹软人脸识别sdk使用工具类

public class FaceUtil { private static final String TAG = FaceUtil.class.getSimpleName(); private static FaceUtil faceInstance = null; public FaceDB mFaceDB; private AFR_FSDKFace mAFR_FSDKFace; private static Context mContext = null; //face detect private AFT_FSDKVersion version = new AFT_FSDKVersion(); private AFT_FSDKEngine engine = new AFT_FSDKEngine(); private ASAE_FSDKVersion mAgeVersion = new ASAE_FSDKVersion(); private ASAE_FSDKEngine mAgeEngine = new ASAE_FSDKEngine(); private ASGE_FSDKVersion mGenderVersion = new ASGE_FSDKVersion(); private ASGE_FSDKEngine mGenderEngine = new ASGE_FSDKEngine(); private List<AFT_FSDKFace> result = new ArrayList<>(); private List<ASAE_FSDKAge> ages = new ArrayList<>(); private List<ASGE_FSDKGender> genders = new ArrayList<>(); private AFT_FSDKFace mAFT_FSDKFace = null; private boolean openFace = false; public static FaceUtil getInstance(Context context) { if(faceInstance == null) { faceInstance = new FaceUtil(context); } return faceInstance; } public FaceUtil(Context context) { mContext = context; } public void setFaceDB() { //face mFaceDB = new FaceDB(mContext.getExternalCacheDir().getPath()); Log.d(TAG,"getExternalCacheDir : "+mContext.getExternalCacheDir().getPath()); } public void initFaceDetect() { AFT_FSDKError err = engine.AFT_FSDK_InitialFaceEngine(FaceDB.appid, FaceDB.ft_key, AFT_FSDKEngine.AFT_OPF_0_HIGHER_EXT, 16, 5); Log.d(TAG, "AFT_FSDK_InitialFaceEngine =" + err.getCode()); err = engine.AFT_FSDK_GetVersion(version); Log.d(TAG, "AFT_FSDK_GetVersion:" + version.toString() + "," + err.getCode()); ASAE_FSDKError error = mAgeEngine.ASAE_FSDK_InitAgeEngine(FaceDB.appid, FaceDB.age_key); Log.d(TAG, "ASAE_FSDK_InitAgeEngine =" + error.getCode()); error = mAgeEngine.ASAE_FSDK_GetVersion(mAgeVersion); Log.d(TAG, "ASAE_FSDK_GetVersion:" + mAgeVersion.toString() + "," + error.getCode()); ASGE_FSDKError error1 = mGenderEngine.ASGE_FSDK_InitgGenderEngine(FaceDB.appid, FaceDB.gender_key); Log.d(TAG, "ASGE_FSDK_InitgGenderEngine =" + error1.getCode()); error1 = mGenderEngine.ASGE_FSDK_GetVersion(mGenderVersion); Log.d(TAG, "ASGE_FSDK_GetVersion:" + mGenderVersion.toString() + "," + error1.getCode()); } public void deInitFaceDetect() { AFT_FSDKError err = engine.AFT_FSDK_UninitialFaceEngine(); Log.d(TAG, "AFT_FSDK_UninitialFaceEngine =" + err.getCode()); ASAE_FSDKError err1 = mAgeEngine.ASAE_FSDK_UninitAgeEngine(); Log.d(TAG, "ASAE_FSDK_UninitAgeEngine =" + err1.getCode()); ASGE_FSDKError err2 = mGenderEngine.ASGE_FSDK_UninitGenderEngine(); Log.d(TAG, "ASGE_FSDK_UninitGenderEngine =" + err2.getCode()); } public boolean faceFeatureDetect(byte[] mYUVData,int width,int height) { boolean bDectectFace = false; AFT_FSDKError err =engine.AFT_FSDK_FaceFeatureDetect(mYUVData, width, height, AFT_FSDKEngine.CP_PAF_NV21, result); Log.d(TAG, "AFT_FSDK_FaceFeatureDetect =" + err.getCode()); Log.d(TAG, "Face=" + result.size()); if (!result.isEmpty()) { for (AFT_FSDKFace face : result) { Log.d(TAG, "Face:" + face.toString()); } mAFT_FSDKFace = result.get(0).clone(); bDectectFace = true; } else { Log.d(TAG,"没有检测到人脸"); bDectectFace = false; } result.clear(); return bDectectFace; } public AFT_FSDKFace getmAFT_FSDKFace() { return mAFT_FSDKFace; } public String getAge(byte[] mYUVData,int width,int height,List<ASAE_FSDKFace> face1) { String age; ASAE_FSDKError error1 = mAgeEngine.ASAE_FSDK_AgeEstimation_Image(mYUVData, width, height, AFT_FSDKEngine.CP_PAF_NV21, face1, ages); Log.d("face", "ASAE_FSDK_AgeEstimation_Image:" + error1.getCode()); if(ages.size()== 0) { Log.d(TAG, "ages.size()== 0" ); age = "年龄未知"; } else { Log.d(TAG, "age:" + ages.get(0).getAge()); age = ages.get(0).getAge() == 0 ? "年龄未知" : ages.get(0).getAge() + "岁"; } return age; } public String getGender(byte[] mYUVData,int width,int height,List<ASGE_FSDKFace> face2) { String gender; ASGE_FSDKError error2 = mGenderEngine.ASGE_FSDK_GenderEstimation_Image(mYUVData, width, height, AFT_FSDKEngine.CP_PAF_NV21, face2, genders); Log.d("face", ",ASGE_FSDK_GenderEstimation_Image:" + error2.getCode()); if(ages.size()== 0) { Log.d(TAG, "ages.size()== 0" ); gender = "性别未知"; } else { Log.d(TAG, "gender:" + genders.get(0).getGender()); gender = genders.get(0).getGender() == -1 ? "性别未知" : (genders.get(0).getGender() == 0 ? "男" : "女"); } return gender; } /** * @param path * @return */ public static Bitmap decodeImage(String path) { Bitmap res; try { ExifInterface exif = new ExifInterface(path); int orientation = exif.getAttributeInt(ExifInterface.TAG_ORIENTATION, ExifInterface.ORIENTATION_NORMAL); BitmapFactory.Options op = new BitmapFactory.Options(); op.inSampleSize = 1; op.inJustDecodeBounds = false; //op.inMutable = true; res = BitmapFactory.decodeFile(path, op); //rotate and scale. Matrix matrix = new Matrix(); if (orientation == ExifInterface.ORIENTATION_ROTATE_90) { matrix.postRotate(90); } else if (orientation == ExifInterface.ORIENTATION_ROTATE_180) { matrix.postRotate(180); } else if (orientation == ExifInterface.ORIENTATION_ROTATE_270) { matrix.postRotate(270); } Bitmap temp = Bitmap.createBitmap(res, 0, 0, res.getWidth(), res.getHeight(), matrix, true); Log.d(TAG, "check target Image:" + temp.getWidth() + "X" + temp.getHeight()); if (!temp.equals(res)) { res.recycle(); } return temp; } catch (Exception e) { e.printStackTrace(); } return null; } public void bitmapDetect(String name,Bitmap mBitmap) { byte[] data = new byte[mBitmap.getWidth() * mBitmap.getHeight() * 3 / 2]; 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(); AFD_FSDKEngine engine = new AFD_FSDKEngine(); AFD_FSDKVersion version = new AFD_FSDKVersion(); List<AFD_FSDKFace> result = new ArrayList<AFD_FSDKFace>(); AFD_FSDKError err = engine.AFD_FSDK_InitialFaceEngine(FaceDB.appid, FaceDB.fd_key, AFD_FSDKEngine.AFD_OPF_0_HIGHER_EXT, 16, 5); Log.d(TAG, "AFD_FSDK_InitialFaceEngine = " + err.getCode()); 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); Log.d(TAG, "MSG_EVENT_FD_ERROR"); ToastUtil.showToast("sdk error :MSG_EVENT_FD_ERROR "+err.getCode()); } err = engine.AFD_FSDK_GetVersion(version); Log.d(TAG, "AFD_FSDK_GetVersion =" + version.toString() + ", " + err.getCode()); err = engine.AFD_FSDK_StillImageFaceDetection(data, mBitmap.getWidth(), mBitmap.getHeight(), AFD_FSDKEngine.CP_PAF_NV21, result); Log.d(TAG, "AFD_FSDK_StillImageFaceDetection =" + err.getCode() + "<" + result.size()); if (!result.isEmpty()) { 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); Log.d(TAG, "AFR_FSDK_InitialEngine = " + error1.getCode()); 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); Log.d(TAG, "MSG_EVENT_FR_ERROR"); ToastUtil.showToast("sdk error :MSG_EVENT_FR_ERROR "+error1.getCode()); } error1 = engine1.AFR_FSDK_GetVersion(version1); Log.d(TAG, "FR=" + version.toString() + "," + error1.getCode()); //(210, 178 - 478, 446), degree = 1 780, 2208 - 1942, 3370 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); Log.d(TAG, "Face=" + result1.getFeatureData()[0] + "," + result1.getFeatureData()[1] + "," + result1.getFeatureData()[2] + "," + error1.getCode()); if(error1.getCode() == error1.MOK) { mAFR_FSDKFace = result1.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); // Canvas face_canvas = new Canvas(face_bitmap); // face_canvas.drawBitmap(mBitmap, result.get(0).getRect(), new Rect(0, 0, width, height), null); // Message reg = Message.obtain(); // reg.what = MSG_CODE; // reg.arg1 = MSG_EVENT_REG; // reg.obj = face_bitmap; // mUIHandler.sendMessage(reg); mFaceDB.addFace(name, mAFR_FSDKFace); Log.d(TAG, "addFace MSG_EVENT_REG jxd"); ToastUtil.showToast(mContext.getString(R.string.register_face_success)+name); } else { // Message reg = Message.obtain(); // reg.what = MSG_CODE; // reg.arg1 = MSG_EVENT_NO_FEATURE; // mUIHandler.sendMessage(reg); Log.d(TAG, "MSG_EVENT_NO_FEATURE"); ToastUtil.showToast(mContext.getString(R.string.no_feature)); } error1 = engine1.AFR_FSDK_UninitialEngine(); Log.d(TAG, "AFR_FSDK_UninitialEngine : " + error1.getCode()); } else { // Message reg = Message.obtain(); // reg.what = MSG_CODE; // reg.arg1 = MSG_EVENT_NO_FACE; // mUIHandler.sendMessage(reg); Log.d(TAG, "MSG_EVENT_NO_FACE"); ToastUtil.showToast(mContext.getString(R.string.no_face)); } err = engine.AFD_FSDK_UninitialFaceEngine(); Log.d(TAG, "AFD_FSDK_UninitialFaceEngine =" + err.getCode()); } public static boolean isExternalStorageDocument(Uri uri) { return "com.android.externalstorage.documents".equals(uri.getAuthority()); } /** * @param uri The Uri to check. * @return Whether the Uri authority is DownloadsProvider. */ public static boolean isDownloadsDocument(Uri uri) { return "com.android.providers.downloads.documents".equals(uri.getAuthority()); } /** * @param uri The Uri to check. * @return Whether the Uri authority is MediaProvider. */ public static boolean isMediaDocument(Uri uri) { return "com.android.providers.media.documents".equals(uri.getAuthority()); } /** * Get the value of the data column for this Uri. This is useful for * MediaStore Uris, and other file-based ContentProviders. * * @param context The context. * @param uri The Uri to query. * @param selection (Optional) Filter used in the query. * @param selectionArgs (Optional) Selection arguments used in the query. * @return The value of the _data column, which is typically a file path. */ public static String getDataColumn(Context context, Uri uri, String selection, String[] selectionArgs) { Cursor cursor = null; final String column = "_data"; final String[] projection = { column }; try { cursor = context.getContentResolver().query(uri, projection, selection, selectionArgs, null); if (cursor != null && cursor.moveToFirst()) { final int index = cursor.getColumnIndexOrThrow(column); return cursor.getString(index); } } finally { if (cursor != null) cursor.close(); } return null; } public String getPath(Uri uri) { if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT) { if (DocumentsContract.isDocumentUri(mContext, uri)) { // ExternalStorageProvider if (isExternalStorageDocument(uri)) { final String docId = DocumentsContract.getDocumentId(uri); final String[] split = docId.split(":"); final String type = split[0]; if ("primary".equalsIgnoreCase(type)) { return Environment.getExternalStorageDirectory() + "http://www.likecs.com/" + split[1]; } // TODO handle non-primary volumes } else if (isDownloadsDocument(uri)) { final String id = DocumentsContract.getDocumentId(uri); final Uri contentUri = ContentUris.withAppendedId( Uri.parse("content://downloads/public_downloads"), Long.valueOf(id)); return getDataColumn(mContext, contentUri, null, null); } else if (isMediaDocument(uri)) { final String docId = DocumentsContract.getDocumentId(uri); final String[] split = docId.split(":"); final String type = split[0]; Uri contentUri = null; if ("image".equals(type)) { contentUri = MediaStore.Images.Media.EXTERNAL_CONTENT_URI; } else if ("video".equals(type)) { contentUri = MediaStore.Video.Media.EXTERNAL_CONTENT_URI; } else if ("audio".equals(type)) { contentUri = MediaStore.Audio.Media.EXTERNAL_CONTENT_URI; } final String selection = "_id=?"; final String[] selectionArgs = new String[] { split[1] }; return getDataColumn(mContext, contentUri, selection, selectionArgs); } } } String[] proj = { MediaStore.Images.Media.DATA }; Cursor actualimagecursor = mContext.getContentResolver().query(uri, proj, null, null, null); int actual_image_column_index = actualimagecursor.getColumnIndexOrThrow(MediaStore.Images.Media.DATA); actualimagecursor.moveToFirst(); String img_path = actualimagecursor.getString(actual_image_column_index); String end = img_path.substring(img_path.length() - 4); if (0 != end.compareToIgnoreCase(".jpg") && 0 != end.compareToIgnoreCase(".png")) { return null; } return img_path; } public void setFaceFunction() { openFace = SharedPreferenceUtil.getBoolean(OPEN_FACE, Constant.FACE, false); openFace = !openFace; SharedPreferenceUtil.putBoolean(OPEN_FACE, Constant.FACE, openFace); if(openFace) { ToastUtil.showToast(R.string.open_face); } else { ToastUtil.showToast(R.string.close_face); } } public boolean getFaceIsOpen() { return openFace; } }

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/zgjzpp.html