mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-17 23:52:45 +00:00
1、目标检测:支持自己训练的模型推理
2、目标检测:支持yolo12模型 3、支持JDK8使用 4、引入离线依赖库,支持完全离线使用 5、优化FaceNet人脸比对速度 6、支持4通道图片检测
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
package cn.smartjavaai.face.config;
|
||||
|
||||
import cn.smartjavaai.face.model.facerec.FaceModel;
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
@@ -21,17 +22,17 @@ public class FaceExtractConfig {
|
||||
private boolean align = true;
|
||||
|
||||
/**
|
||||
* 人脸检测模型配置
|
||||
* 人脸检测模型
|
||||
*/
|
||||
private FaceModelConfig detectModelConfig;
|
||||
private FaceModel detectModel;
|
||||
|
||||
public FaceExtractConfig() {
|
||||
}
|
||||
|
||||
public FaceExtractConfig(boolean cropFace, boolean align, FaceModelConfig detectModelConfig) {
|
||||
public FaceExtractConfig(boolean cropFace, boolean align, FaceModel detectModel) {
|
||||
this.cropFace = cropFace;
|
||||
this.align = align;
|
||||
this.detectModelConfig = detectModelConfig;
|
||||
this.detectModel = detectModel;
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -52,6 +52,11 @@ public class FaceModelConfig {
|
||||
*/
|
||||
private int gpuId = 0;
|
||||
|
||||
/**
|
||||
* 人脸特征提取配置
|
||||
*/
|
||||
private FaceExtractConfig extractConfig;
|
||||
|
||||
public FaceModelConfig() {
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user