gitee/face-search
1
0
Fork 0
mirror of https://gitee.com/open-visual/face-search.git synced 2025-07-25 19:41:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

update:添加SeetaFace6Open的人脸特征提起模型

Browse Source
This commit is contained in:
divenswu 2023-04-24 17:28:50 +08:00
parent 5e9348e59c
commit 4eca8aa2cf
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
face-search-core/src/main/resources/model/onnx/recognition_fcae_seeta/face_recognizer_512.onnx → face-search-core/src/main/resources/model/onnx/recognition_face_seeta/face_recognizer_512.onnx
View File

4
face-search-server/src/main/java/com/visual/face/search/server/bootstrap/conf/ModelConfig.java
View File

@ -202,6 +202,10 @@ public class ModelConfig {
return new String[]{basePath + "model/onnx/recognition_face_arc/glint360k_cosface_r18_fp16_0.1.onnx"}; 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)){ if((null == modelPath || modelPath.length != 1) && "InsightAttributeDetection".equalsIgnoreCase(modelName)){
return new String[]{basePath + "model/onnx/attribute_gender_age/insight_gender_age.onnx"}; return new String[]{basePath + "model/onnx/attribute_gender_age/insight_gender_age.onnx"};
} }