集成算法seetaface6

2026-09-20 01:29:18 +00:00 · 2025-03-26 17:09:22 +08:00
parent 45cb7a9edc
commit 72a4c2c058
78 changed files with 1843 additions and 264 deletions
--- a/smartjavaai-face/src/main/java/com/seetaface/SeetaFace6JNI.java
+++ b/smartjavaai-face/src/main/java/com/seetaface/SeetaFace6JNI.java
@@ -0,0 +1,127 @@
+package com.seetaface;
+
+
+import com.seetaface.model.RecognizeResult;
+import com.seetaface.model.SeetaImageData;
+import com.seetaface.model.SeetaPointF;
+import com.seetaface.model.SeetaRect;
+
+/**
+ * seetaface6 sdk
+ * @author dwj
+ */
+public class SeetaFace6JNI {
+
+    /**
+     * 初始化，指定人脸识别模型文件目录
+     *
+     * @param modelDir
+     * @return
+     */
+
+    public native boolean initModel(String modelDir);
+
+    /**
+     * 检测人脸
+     *
+     * @param img
+     * @return
+     */
+    public native SeetaRect[] detect(SeetaImageData img);
+
+    /**
+     * 根据人脸检测关键点
+     * 关键定定位输入的是原始图片和人脸检测结果，给出指定人脸上的关键点的依次坐标。
+     * 这里检测到的5点坐标循序依次为，左眼中心、右眼中心、鼻尖、左嘴角和右嘴角。
+     * 注意这里的左右是基于图片内容的左右，并不是图片中人的左右，即左眼中心就是图片中左边的眼睛的中心。
+     *
+     * @param img
+     * @param faces
+     * @return
+     */
+    public native SeetaPointF[] mark(SeetaImageData img, SeetaRect faces);
+
+    /**
+     * 1 v 1 人脸比对
+     *
+     * @param img1
+     * @param img2
+     * @return 相似度范围在0~1,返回负数表示出错
+     */
+    public native float compare(SeetaImageData img1, SeetaImageData img2);
+
+    /**
+     * 提取人脸区域特性
+     * @param face crop方法返回的人脸图像
+     * @return
+     */
+    public native float[] extractCroppedFace(byte[] face);
+
+    /**
+     * 提取一个图像中最大人脸的特征
+     * @param img
+     * @return
+     */
+    public native float[] extractMaxFace(SeetaImageData img);
+
+    /**
+     * 计算两个特性的相似度
+     * @param features1
+     * @param features2
+     * @return
+     */
+    public native float calculateSimilarity(float[] features1, float[] features2);
+
+    /**
+     * 注册人脸
+     *
+     * @param img
+     * @return The returned value is the index of face database. Reture -1 if failed
+     */
+    public native long register(SeetaImageData img);
+
+    /**
+     * 注册裁剪后的人脸，推荐使用该方法
+     * @param bytes
+     * @return
+     */
+    public native long registerCroppedFace(byte[] bytes);
+
+    /**
+     * 从人脸库中搜索，返回相似度最高的索引
+     *
+     * @param img
+     * @return index saves the index of face databese, which is same as the retured value by Register. similar saves the most similar.
+     */
+    public native RecognizeResult query(SeetaImageData img);
+
+    /**
+     * 用裁剪后的人脸进行搜索
+     * @param bytes
+     * @return
+     */
+    public native RecognizeResult queryByCroppedFace(byte[] bytes);
+
+    /**
+     * 将人脸从数据库中删除
+     * @param index -1: 删除所有
+     * @return 返回删除记录数
+     */
+    public native long delete(long[] index);
+
+    /**
+     * 人脸提取
+     *
+     * @param img
+     * @return The returned value is face data. Reture null if failed
+     */
+    public native byte[][] crop(SeetaImageData img);
+
+    /**
+     * 图片活体检测
+     * @param img
+     * @return
+     */
+    public native int predictImage(SeetaImageData img);
+
+}