From 7ba02e12d8e1177dd0041491a6179db08862f8e0 Mon Sep 17 00:00:00 2001
From: dengwenjie <775747758@qq.com>
Date: Sat, 17 May 2025 13:32:04 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96demo?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/smartai/examples/face/facerec/FaceNetDemo.java  | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/examples/src/main/java/smartai/examples/face/facerec/FaceNetDemo.java b/examples/src/main/java/smartai/examples/face/facerec/FaceNetDemo.java
index 67bf62a..eaaec70 100644
--- a/examples/src/main/java/smartai/examples/face/facerec/FaceNetDemo.java
+++ b/examples/src/main/java/smartai/examples/face/facerec/FaceNetDemo.java
@@ -125,7 +125,7 @@ public class FaceNetDemo {
         try {
             FaceModelConfig config = new FaceModelConfig();
             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);
             //自动裁剪人脸并比对人脸特征
             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();
             config.setModelEnum(FaceModelEnum.FACENET_FEATURE_EXTRACTION);//人脸模型
             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);
             //自动裁剪人脸并比对人脸特征
             float similar = faceModel.featureComparison("src/main/resources/kana1.jpg","src/main/resources/kana2.jpg");