ZF/SmartJavaAI
1
0
Fork 0
mirror of https://github.com/geekwenjie/SmartJavaAI.git synced 2026-09-14 05:38:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

1、人脸模块：新增小视科技（MiniVision）活体检测模型

Browse Source 
2、人脸模块：新增阿里通义工作室活体检测模型
3、人脸模块：新增2个表情识别模型
4、人脸模块：新增InsightFace、ElasticFace人脸识别模型
5、人脸模块：新增Seetaface6质量评估模型
6、目标检测模块：开放更多自定义模型参数
7、人脸模块：支持base64图片
8、实现接口 AutoCloseable，支持资源的自动释放
9、OCR模块：解决加方向矫正后无法连续识别bug
10、人脸模块：解决人脸更新后缓存问题
11、优化部分功能
This commit is contained in:
dengwenjie
2025-07-07 08:45:08 +08:00
parent 3e631a060b
commit 07a8a18835
168 changed files with 8562 additions and 2850 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
smartjavaai-face/src/main/java/cn/smartjavaai/face/utils/FaceUtils.java
View File

@@ -7,11 +7,14 @@ import ai.djl.ndarray.NDArray;
import ai.djl.ndarray.NDManager;
import cn.smartjavaai.common.entity.*;
import cn.smartjavaai.common.entity.Point;
import cn.smartjavaai.common.enums.EyeStatus;
import cn.smartjavaai.common.enums.GenderType;
import cn.smartjavaai.common.entity.face.FaceAttribute;
import cn.smartjavaai.common.entity.face.FaceInfo;
import cn.smartjavaai.common.entity.face.HeadPose;
import cn.smartjavaai.common.entity.face.LivenessResult;
import cn.smartjavaai.common.enums.face.EyeStatus;
import cn.smartjavaai.common.enums.face.GenderType;
import cn.smartjavaai.common.utils.ImageUtils;
import cn.smartjavaai.face.config.FaceModelConfig;
import cn.smartjavaai.common.enums.LivenessStatus;
import cn.smartjavaai.common.enums.face.LivenessStatus;
import cn.smartjavaai.face.exception.FaceException;
import com.seeta.sdk.*;
@@ -340,6 +343,44 @@ public class FaceUtils {
return points;
}
/**
* 112x112的目标点 - Target point of 112x112
* standard 5 landmarks for FFHQ faces with 112x112
*
* @param manager
* @return
*/
public static NDArray faceTemplate112x112(NDManager manager) {
double[][] coord5point = {
{30.29459953, 51.69630051}, // 112x112的目标点 - Target point of 512x512
{65.53179932, 51.50139999},
{48.02519989, 71.73660278},
{33.54930115, 87},
{62.72990036, 87}
};
NDArray points = manager.create(coord5point);
return points;
}
/**
* 96x112的目标点 - Target point of 96x112
* standard 5 landmarks for FFHQ faces with 96x112
*
* @param manager
* @return
*/
public static NDArray faceTemplate96x112(NDManager manager) {
double[][] coord5point = {
{30.29459953, 51.69630051},
{65.53179932, 51.50139999},
{48.02519989, 71.73660278},
{33.54930115, 92.3655014},
{62.72990036, 92.20410156}
};
NDArray points = manager.create(coord5point);
return points;
}
/**
* bgr转图片
* @return 图片
@@ -507,7 +548,7 @@ public class FaceUtils {
.map(p -> new Point(p.x, p.y))
.collect(Collectors.toList());
FaceInfo faceInfo = new FaceInfo(keyPoints);
faceInfo.setLivenessStatus(livenessStatusList.get(i));
faceInfo.setLivenessStatus(new LivenessResult(livenessStatusList.get(i)));
DetectionInfo detectionInfo = new DetectionInfo(rectangle, 0, faceInfo);
detectionInfoList.add(detectionInfo);
}