mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-15 22:57:26 +00:00
136 lines
3.3 KiB
Java
136 lines
3.3 KiB
Java
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);
|
||
|
||
public native void dispose();
|
||
|
||
@Override
|
||
protected void finalize() throws Throwable {
|
||
super.finalize();
|
||
this.dispose();
|
||
}
|
||
|
||
}
|