From 4eca8aa2cf2773c5d38c5b36d0fbdeeb9b1e4d16 Mon Sep 17 00:00:00 2001 From: divenswu Date: Mon, 24 Apr 2023 17:28:50 +0800 Subject: [PATCH] =?UTF-8?q?update:=E6=B7=BB=E5=8A=A0SeetaFace6Open?= =?UTF-8?q?=E7=9A=84=E4=BA=BA=E8=84=B8=E7=89=B9=E5=BE=81=E6=8F=90=E8=B5=B7?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../face_recognizer_512.onnx | Bin .../search/server/bootstrap/conf/ModelConfig.java | 4 ++++ 2 files changed, 4 insertions(+) rename face-search-core/src/main/resources/model/onnx/{recognition_fcae_seeta => recognition_face_seeta}/face_recognizer_512.onnx (100%) diff --git a/face-search-core/src/main/resources/model/onnx/recognition_fcae_seeta/face_recognizer_512.onnx b/face-search-core/src/main/resources/model/onnx/recognition_face_seeta/face_recognizer_512.onnx similarity index 100% rename from face-search-core/src/main/resources/model/onnx/recognition_fcae_seeta/face_recognizer_512.onnx rename to face-search-core/src/main/resources/model/onnx/recognition_face_seeta/face_recognizer_512.onnx diff --git a/face-search-server/src/main/java/com/visual/face/search/server/bootstrap/conf/ModelConfig.java b/face-search-server/src/main/java/com/visual/face/search/server/bootstrap/conf/ModelConfig.java index 27cba1e..fe03270 100755 --- a/face-search-server/src/main/java/com/visual/face/search/server/bootstrap/conf/ModelConfig.java +++ b/face-search-server/src/main/java/com/visual/face/search/server/bootstrap/conf/ModelConfig.java @@ -202,6 +202,10 @@ public class ModelConfig { return new String[]{basePath + "model/onnx/recognition_face_arc/glint360k_cosface_r18_fp16_0.1.onnx"}; } + if((null == modelPath || modelPath.length != 1) && "SeetaFaceOpenRecognition".equalsIgnoreCase(modelName)){ + return new String[]{basePath + "model/onnx/recognition_face_seeta/face_recognizer_512.onnx"}; + } + if((null == modelPath || modelPath.length != 1) && "InsightAttributeDetection".equalsIgnoreCase(modelName)){ return new String[]{basePath + "model/onnx/attribute_gender_age/insight_gender_age.onnx"}; }