修复人脸识别算法facenet-pytorch实现方式

smartjavaai-face/src/main/java/cn/smartjavaai/face/translator/FaceFeatureTranslator.java

@@ -0,0 +1,52 @@
+package cn.smartjavaai.face.translator;
+
+import ai.djl.modality.cv.Image;
+import ai.djl.modality.cv.transform.Normalize;
+import ai.djl.modality.cv.transform.Resize;
+import ai.djl.modality.cv.transform.ToTensor;
+import ai.djl.ndarray.NDArray;
+import ai.djl.ndarray.NDList;
+import ai.djl.translate.Batchifier;
+import ai.djl.translate.Pipeline;
+import ai.djl.translate.Translator;
+import ai.djl.translate.TranslatorContext;
+
+/**
+ * @author 邓文杰
+ * @date 2025/3/31
+ */
+public final class FaceFeatureTranslator implements Translator<Image, float[]> {
+
+    public FaceFeatureTranslator() {
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    public NDList processInput(TranslatorContext ctx, Image input) {
+        NDArray array = input.toNDArray(ctx.getNDManager(), Image.Flag.COLOR);
+        Pipeline pipeline = new Pipeline();
+        pipeline
+                .add(new Resize(180))
+                .add(new ToTensor())
+                .add(new Normalize(
+                        new float[]{127.5f / 255.0f, 127.5f / 255.0f, 127.5f / 255.0f},
+                        new float[]{128.0f / 255.0f, 128.0f / 255.0f, 128.0f / 255.0f}));
+
+        return pipeline.transform(new NDList(array));
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    public float[] processOutput(TranslatorContext ctx, NDList list) {
+        return list.singletonOrThrow().toFloatArray();
+    }
+
+    @Override
+    public Batchifier getBatchifier() {
+        return Batchifier.STACK;
+    }
+}