ZF/SmartJavaAI
1
0
Fork 0
mirror of https://github.com/geekwenjie/SmartJavaAI.git synced 2026-09-14 13:48:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

优化demo

Browse Source
This commit is contained in:
dengwenjie
2025-05-17 13:32:04 +08:00
parent a7a118c5aa
commit 7ba02e12d8
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/src/main/java/smartai/examples/face/facerec/FaceNetDemo.java
View File

@@ -125,7 +125,7 @@ public class FaceNetDemo {
try { try {
FaceModelConfig config = new FaceModelConfig(); FaceModelConfig config = new FaceModelConfig();
config.setModelEnum(FaceModelEnum.FACENET_FEATURE_EXTRACTION);//人脸模型 config.setModelEnum(FaceModelEnum.FACENET_FEATURE_EXTRACTION);//人脸模型
//config.setModelPath("/Users/xxx/Documents/develop/face_model/model_ir_se50.pth"); config.setModelPath("/Users/wenjie/Documents/develop/face_model/face_feature.pt");
FaceModel faceModel = FaceModelFactory.getInstance().getModel(config); FaceModel faceModel = FaceModelFactory.getInstance().getModel(config);
//自动裁剪人脸并比对人脸特征 //自动裁剪人脸并比对人脸特征
float similar = faceModel.featureComparison("src/main/resources/kana1.jpg","src/main/resources/kana2.jpg"); float similar = faceModel.featureComparison("src/main/resources/kana1.jpg","src/main/resources/kana2.jpg");
@@ -147,6 +147,13 @@ public class FaceNetDemo {
FaceModelConfig config = new FaceModelConfig(); FaceModelConfig config = new FaceModelConfig();
config.setModelEnum(FaceModelEnum.FACENET_FEATURE_EXTRACTION);//人脸模型 config.setModelEnum(FaceModelEnum.FACENET_FEATURE_EXTRACTION);//人脸模型
config.setModelPath("/Users/xxx/Documents/develop/face_model/face_feature.pt"); config.setModelPath("/Users/xxx/Documents/develop/face_model/face_feature.pt");
//人脸特征提取参数
FaceExtractConfig extractConfig = new FaceExtractConfig();
FaceModelConfig detectModelConfig = new FaceModelConfig(FaceModelEnum.ULTRA_LIGHT_FAST_GENERIC_FACE);
detectModelConfig.setModelPath("/Users/xxx/Documents/develop/face_model/ultranet.pt");
//人脸检测模型配置
extractConfig.setDetectModel(FaceModelFactory.getInstance().getModel(detectModelConfig));
config.setExtractConfig(extractConfig);
FaceModel faceModel = FaceModelFactory.getInstance().getModel(config); FaceModel faceModel = FaceModelFactory.getInstance().getModel(config);
//自动裁剪人脸并比对人脸特征 //自动裁剪人脸并比对人脸特征
float similar = faceModel.featureComparison("src/main/resources/kana1.jpg","src/main/resources/kana2.jpg"); float similar = faceModel.featureComparison("src/main/resources/kana1.jpg","src/main/resources/kana2.jpg");