SmartJavaAI/smartjavaai-face/src/main/java/cn/smartjavaai/face/config/FaceModelConfig.java

package cn.smartjavaai.face.config;

import cn.smartjavaai.common.enums.DeviceEnum;
import cn.smartjavaai.face.constant.FaceDetectConstant;
import cn.smartjavaai.face.enums.FaceModelEnum;
import cn.smartjavaai.face.enums.VectorDBType;
import cn.smartjavaai.face.vector.config.VectorDBConfig;
import lombok.Data;

/**
 * 人脸检测识别模型配置
 * @author dwj
 */
@Data
public class FaceModelConfig {

    /**
     * 人脸模型枚举
     */
    private FaceModelEnum modelEnum;

    /**
     * 置信度阈值
     */
    private double confidenceThreshold = FaceDetectConstant.DEFAULT_CONFIDENCE_THRESHOLD;

    /**
     * 相似度阈值 作用：判断是否为同一人脸
     */
    //private double similarityThreshold = 0D;

    /**
     * 非极大抑制阈值 作用：消除重叠检测框，保留最优结果
     */
    private double nmsThresh = FaceDetectConstant.NMS_THRESHOLD;

    /**
     * 模型路径
     */
    private String modelPath;

    /**
     * 人脸库路径
     */
    private String faceDbPath;

    /**
     * 设备类型
     */
    private DeviceEnum device;

    /**
     * gpu设备ID 当device为GPU时生效
     */
    private int gpuId = 0;

    /**
     * 人脸特征提取配置
     */
    private FaceExtractConfig extractConfig;


    /**
     * 向量数据库配置
     */
    private VectorDBConfig vectorDBConfig;

    /**
     * 是否自动加载人脸到内存
     */
    private boolean isAutoLoadFace = true;

    public FaceModelConfig() {
    }

    public FaceModelConfig(FaceModelEnum modelEnum) {
        this.modelEnum = modelEnum;
    }

    public FaceModelConfig(FaceModelEnum modelEnum, String modelPath) {
        this.modelEnum = modelEnum;
        this.modelPath = modelPath;
    }
}