package com.example.datalibrary.manager;
|
|
import android.content.Context;
|
|
import com.baidu.idl.main.facesdk.FaceCrop;
|
import com.baidu.idl.main.facesdk.FaceDarkEnhance;
|
import com.baidu.idl.main.facesdk.FaceDetect;
|
import com.baidu.idl.main.facesdk.FaceFeature;
|
import com.baidu.idl.main.facesdk.FaceLive;
|
import com.baidu.idl.main.facesdk.FaceMouthMask;
|
import com.baidu.idl.main.facesdk.FaceSearch;
|
import com.baidu.idl.main.facesdk.model.BDFaceInstance;
|
import com.baidu.idl.main.facesdk.model.BDFaceSDKConfig;
|
import com.example.datalibrary.factory.specific.CropBuilder;
|
import com.example.datalibrary.factory.specific.DarkBuilder;
|
import com.example.datalibrary.factory.specific.DetectBuilder;
|
import com.example.datalibrary.factory.specific.DetectNirBuilder;
|
import com.example.datalibrary.factory.specific.DetectQualityBuilder;
|
import com.example.datalibrary.factory.specific.FeatureBuilder;
|
import com.example.datalibrary.factory.specific.LiveBuilder;
|
import com.example.datalibrary.factory.specific.MouthMaskBuilder;
|
import com.example.datalibrary.factory.specific.TrackBuilder;
|
import com.example.datalibrary.listener.SdkInitListener;
|
|
public class FaceModel implements SdkInitListener{
|
private TrackBuilder trackBuilder;
|
private DetectBuilder detectBuilder;
|
private DetectQualityBuilder detectQualityBuilder;
|
private DetectNirBuilder detectNirBuilder;
|
private DarkBuilder darkBuilder;
|
private FeatureBuilder featureBuilder;
|
private FeatureBuilder featurePersonBuilder;
|
private LiveBuilder liveBuilder;
|
private CropBuilder cropBuilder;
|
private MouthMaskBuilder mouthMaskBuilder = null;
|
private boolean isModelInit;
|
private boolean checkMouthMask = false;
|
|
public void setListener(SdkInitListener listener) {
|
this.listener = listener;
|
}
|
|
private SdkInitListener listener;
|
public FaceModel(boolean checkMouthMask){
|
isModelInit = false;
|
cropBuilder = new CropBuilder(this);
|
trackBuilder = new TrackBuilder(this);
|
detectBuilder = new DetectBuilder(this);
|
detectQualityBuilder = new DetectQualityBuilder(this);
|
detectNirBuilder = new DetectNirBuilder(this);
|
darkBuilder = new DarkBuilder(this);
|
liveBuilder = new LiveBuilder(this);
|
featureBuilder = new FeatureBuilder(this);
|
featurePersonBuilder = new FeatureBuilder(this);
|
|
this.checkMouthMask = checkMouthMask;
|
if (checkMouthMask) {
|
mouthMaskBuilder = new MouthMaskBuilder(this);
|
}
|
}
|
public void init(BDFaceSDKConfig config , Context context){
|
if (isModelInit){
|
return;
|
}
|
BDFaceInstance trackInstance = new BDFaceInstance();
|
trackInstance.creatInstance();
|
BDFaceInstance detectInstance = new BDFaceInstance();
|
detectInstance.creatInstance();
|
// 人证核验init入参
|
BDFaceInstance detectQualityInstance = new BDFaceInstance();
|
detectQualityInstance.creatInstance();
|
BDFaceInstance cropInstance = new BDFaceInstance();
|
cropInstance.creatInstance();
|
|
trackBuilder.init(trackInstance , config);
|
cropBuilder.init(cropInstance);
|
detectBuilder.init(detectInstance , config);
|
detectNirBuilder.init(detectInstance);
|
// 认证核验检测模型预配置
|
detectQualityBuilder.init(detectQualityInstance , config);
|
darkBuilder.init(null);
|
liveBuilder.init(null);
|
// 认证核验提取特征模型预配置
|
featurePersonBuilder.init(detectQualityInstance);
|
// 通用特征提取模型预配置
|
featureBuilder.init(null);
|
|
if (checkMouthMask){
|
mouthMaskBuilder.init(null);
|
}
|
|
if (checkMouthMask) {
|
// 口罩模型初始化
|
mouthMaskBuilder.initModel(context);
|
}
|
// 抠图模型初始化
|
cropBuilder.initModel(context);
|
// 跟踪模型初始化
|
trackBuilder.initModel(context);
|
// 检测模型初始化
|
detectBuilder.initModel(context);
|
// 认证核验检测模型初始化 （边导入边识别-需初始化此处）
|
detectQualityBuilder.initModel(context);
|
// 红外检测模型
|
detectNirBuilder.initModel(context);
|
// 暗光恢复模型
|
darkBuilder.initModel(context);
|
// 活体分数模型
|
liveBuilder.initModel(context);
|
// 认证特征提取模型 （边导入边识别-需初始化此处）
|
featurePersonBuilder.initModel(context);
|
// 通用特征提取模型
|
featureBuilder.initModel(context);
|
}
|
|
public void setCheckMouthMask(boolean checkMouthMask){
|
this.checkMouthMask = checkMouthMask;
|
}
|
|
@Override
|
public void initStart() {
|
|
listener.initStart();
|
}
|
|
@Override
|
public void initLicenseSuccess() {
|
|
listener.initLicenseSuccess();
|
}
|
|
@Override
|
public void initLicenseFail(int errorCode, String msg) {
|
listener.initLicenseFail(errorCode , msg);
|
}
|
|
@Override
|
public void initModelSuccess() {
|
listener.initModelSuccess();
|
isModelInit = true;
|
}
|
|
@Override
|
public void initModelFail(int errorCode, String msg) {
|
isModelInit = false;
|
listener.initModelFail(errorCode , msg);
|
}
|
public FaceFeature getFacePersonFeature(){
|
return featurePersonBuilder.getExample();
|
}
|
public FaceSearch getFacePersonSearch(){
|
return featurePersonBuilder.getFaceSearch();
|
}
|
public FaceFeature getFaceFeature(){
|
return featureBuilder.getExample();
|
}
|
public FaceSearch getFaceSearch(){
|
return featureBuilder.getFaceSearch();
|
}
|
public FaceDetect getFaceTrack(){
|
return trackBuilder.getExample();
|
}
|
public FaceCrop getFaceCrop(){
|
return cropBuilder.getExample();
|
}
|
public FaceDetect getFaceDetectPerson(){
|
return detectQualityBuilder.getExample();
|
}
|
public FaceDetect getFaceDetect(){
|
return detectBuilder.getExample();
|
}
|
public FaceDetect getFaceNirDetect(){
|
return detectNirBuilder.getExample();
|
}
|
public FaceDarkEnhance getDark(){
|
return darkBuilder.getExample();
|
}
|
public FaceLive getFaceLive(){
|
return liveBuilder.getExample();
|
}
|
|
public FaceMouthMask getFaceMoutMask(){
|
if (mouthMaskBuilder != null) {
|
return mouthMaskBuilder.getExample();
|
}
|
else{
|
return null;
|
}
|
}
|
}
|
