初始提交

smartjavaai-face/src/main/java/cn/smartjavaai/face/FaceAlgorithm.java

@@ -0,0 +1,83 @@
+package cn.smartjavaai.face;
+
+import ai.djl.MalformedModelException;
+import ai.djl.repository.zoo.ModelNotFoundException;
+import ai.djl.translate.TranslateException;
+
+import java.io.IOException;
+import java.io.InputStream;
+
+/**
+ * 人脸识别算法
+ * @author dwj
+ * @date 2025/2/19
+ */
+public interface FaceAlgorithm {
+
+    /**
+     * 加载模型
+     * @param config
+     * @throws Exception
+     */
+    void loadModel(ModelConfig config) throws Exception; // 加载模型
+
+    /**
+     * 人脸检测
+     * @param imagePath 图片路径
+     * @return
+     * @throws Exception
+     */
+    FaceDetectedResult detect(String imagePath) throws Exception;
+
+    /**
+     * 人脸检测
+     * @param imageInputStream 图片输入流
+     * @return
+     * @throws Exception
+     */
+    FaceDetectedResult detect(InputStream imageInputStream) throws Exception;
+
+    /**
+     * 特征提取
+     * @param imagePath 图片路径
+     * @return
+     * @throws Exception
+     */
+    float[] featureExtraction(String imagePath) throws Exception;
+
+    /**
+     * 特征提取
+     * @param inputStream 输入流
+     * @return
+     * @throws Exception
+     */
+    float[] featureExtraction(InputStream inputStream) throws Exception;
+
+    /**
+     * 计算相似度
+     * @param feature1 图1特征
+     * @param feature2 图2特征
+     * @return
+     * @throws Exception
+     */
+    float calculSimilar(float[] feature1, float[] feature2) throws Exception;
+
+    /**
+     * 特征比较
+     * @param imagePath1 图1路径
+     * @param imagePath2 图2路径
+     * @return
+     * @throws Exception
+     */
+    float featureComparison(String imagePath1, String imagePath2) throws Exception;
+
+    /**
+     * 特征比较
+     * @param inputStream1 图1输入流
+     * @param inputStream2 图2输入流
+     * @return
+     * @throws Exception
+     */
+    float featureComparison(InputStream inputStream1, InputStream inputStream2) throws Exception;
+
+}